set_data_type
功能说明
设置张量数据类型。
函数原型
1 | set_data_type(data_type: DataType) -> TensorHolder |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
data_type |
输入 |
数据类型枚举。详见DataType。 |
返回值说明
(TensorHolder) 返回自身以支持链式调用。
约束说明
- 如果data_type不是DataType枚举，抛出TypeError。
- 如果操作失败，抛出RuntimeError。
父主题： TensorHolder