Atlas 训练系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 推理系列产品
将输入tensor转换为指定的dtype类型。
const aclTensor *Cast(const aclTensor *self, op::DataType dstDtype, aclOpExecutor *executor)
参数 |
输入/输出 |
说明 |
|---|---|---|
self |
输入 |
需要被转换的tensor，数据类型支持FLOAT16、FLOAT、DOUBLE、BFLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、COMPLEX64、COMPLEX128。数据格式支持ND。 说明：
其中BFLOAT16适用于如下产品型号： Atlas A2训练系列产品/Atlas 800I A2推理产品 |
dstDtype |
输入 |
输入tensor要转换的目标dtype，数据类型支持FLOAT16、FLOAT、DOUBLE、BFLOAT16、、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、COMPLEX64、COMPLEX128。 说明：
其中BFLOAT16适用于如下产品型号： Atlas A2训练系列产品/Atlas 800I A2推理产品 |
executor |
输入 |
op执行器，包含了算子计算流程。 |
返回类型为dstDtype的tensor。
无