---
title: 简介
description: "InferDataTypeContext继承自ExtendedKernelContext，是一个用于数据类型（DataType）推导的上下文类，该类的主要作用是在算子数据类型推导过程中，提供获取输入、输出数据类型的接口，并支持设置输出数据类型。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/maintenref/basicdataapi/atlasopapi_07_00547.html
sourcePath: /source/zh/canncommercial/900/maintenref/basicdataapi/atlasopapi_07_00547.html
indexId: 8444825d654c4737a981979adc6d70fa63f05dec8cff183708c00770c2cedd0f78
---
# 简介

InferDataTypeContext继承自ExtendedKernelContext，是一个用于数据类型（DataType）推导的上下文类，该类的主要作用是在算子数据类型推导过程中，提供获取输入、输出数据类型的接口，并支持设置输出数据类型。

InferDataTypeContext继承关系图如下：


#### 需要包含的头文件

```
#include <infer_datatype_context.h>
```


#### Public成员函数

```
ge::DataType
GetInputDataType
(const size_t index) const
ge::DataType
GetOptionalInputDataType
(const size_t ir_index) const
ge::DataType
GetDynamicInputDataType
(const size_t ir_index, const size_t relative_index) const
ge::DataType
GetRequiredInputDataType
(const size_t ir_index) const
ge::DataType
GetOutputDataType
(const size_t index) const
ge::graphStatus
SetOutputDataType
(const size_t index, const
ge::DataType
datatype)
```