SetTensorValue

设置Tensor的值，支持int32_t、uint8_t、float16、float32类型。

Tensor对象需在Device侧且数据类型与调用的SetTensorValue方法匹配。

Tensor所在的Device需与stream所在的Device一致。

APP_ERROR SetTensorValue(uint8_t value, AscendStream& stream = AscendStream::DefaultStream());

APP_ERROR SetTensorValue(float value, bool IsFloat16 = false, AscendStream& stream = AscendStream::DefaultStream());

APP_ERROR SetTensorValue(int32_t value, AscendStream& stream = AscendStream::DefaultStream());

参数名	输入/输出	说明
value	输入	int32_t、uint8_t、float类型，指定Tensor待设置的值。
isFloat16	输入	bool类型，默认为“false”。为“true”时表示设置Tensor元素为floa16类型。为“false”时表示设置Tensor元素为floa32类型。
stream	输入	AscendStream类型，默认值为AscendStream::DefaultStream()。当参数值为默认值时，接口为同步操作，其他情况下，接口为异步操作。

参数名

输入/输出

说明

value

输入

int32_t、uint8_t、float类型，指定Tensor待设置的值。

isFloat16

输入

bool类型，默认为“false”。

stream

输入

AscendStream类型，默认值为AscendStream::DefaultStream()。当参数值为默认值时，接口为同步操作，其他情况下，接口为异步操作。

数据结构	说明
APP_ERROR	程序执行返回的错误码，请参考APP_ERROR说明。

父主题： Tensor