SetSize

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

功能说明

设置当前LocalTensor Size大小。单位为元素。当用户重用local tensor变量且使用长度发生变化的时候，需要使用此接口重新设置Size。

函数原型

__aicore__ inline void SetSize(const uint32_t size)

参数说明

表 1 参数说明

参数名	输入/输出	描述
size	输入	元素个数，单位为元素。

返回值说明

无

约束说明

无

调用示例

参考调用示例。