---
title: InferDataType
description: "注册算子的InferDataType函数。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/maintenref/basicdataapi/atlasopapi_07_00117.html
sourcePath: /source/zh/canncommercial/900/maintenref/basicdataapi/atlasopapi_07_00117.html
indexId: cf921638cc17a9afb503e6c37782ddb944053f24f1661e0f7978cc356cd52ecd78
---
# InferDataType

#### 函数功能

注册算子的InferDataType函数。

用户需要为算子编写一个InferDataTypeKernelFunc类型的函数，并使用该接口进行注册。

InferDataTypeKernelFunc类型定义如下：

```
using InferDataTypeKernelFunc = UINT32 (*)(InferDataTypeContext *);
```


#### 函数原型

```
OpImplRegisterV2 &InferDataType(InferDataTypeKernelFunc infer_datatype_func)
```


#### 参数说明

| 参数 | 输入/输出 | 说明 |
| --- | --- | --- |
| infer\_datatype\_func | 输入 | 要注册的自定义InferDataType函数，类型为InferDataTypeKernelFunc。 |


#### 返回值说明

返回算子的OpImplRegisterV2对象，该对象新增注册了InferDataType函数infer_datatype_func。


#### 约束说明

无。