set_tensor_value
函数功能
设置Tensor的值,支持dtype.int32、dtype.uint8、dtype.float16、dtype.float32类型。
- 请注意数据类型与数据范围。
- Tensor对象需在Device侧且数据类型与调用的set_tensor_value的方法匹配。
函数原型
set_tensor_value(value: float, dataType: dtype)
输入参数说明
参数名 |
类型 |
说明 |
---|---|---|
value |
float |
需要设置的值,不可为空。 |
dataType |
dtype枚举类型 |
dataType: dtype 转换类型,不可为空,支持的输入有:dtype.float16,dtype.float32,dtype.uint8,dtype.int32。 |
返回参数说明
返回进行张量设置值后的Tensor类数据。
父主题: Tensor