set_data_type

产品支持情况

产品	是否支持
Atlas 350 加速卡	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

设置张量数据类型。

函数原型

set_data_type(data_type: DataType) -> Tensor

参数说明

参数名	输入/输出	描述
data_type	输入	要设置的数据类型。详细类型请参见DataType。

返回值说明

(Tensor) 返回自身以支持链式调用。

约束说明

如果data_type不是DataType类型，抛出TypeError。
如果设置失败，抛出RuntimeError。

父主题： Tensor