昇腾社区首页
中文
注册

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类数据。