set_data_type

功能说明

设置张量数据类型。

函数原型

1

set_data_type(data_type: DataType) -> Tensor

参数说明

返回值说明

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

约束说明

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