SetSize
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
设置当前LocalTensor Size大小。单位为元素。当用户重用local tensor变量且使用长度发生变化的时候,需要使用此接口重新设置Size。
函数原型
1 | __aicore__ inline void SetSize(const uint32_t size) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
size |
输入 |
元素个数,单位为元素。 |
返回值说明
无
约束说明
无
调用示例
// 示例1 SetSize 如下示例将申请的Tensor长度修改为256(单位为元素) AscendC::LocalTensor<float> tmpBuffer3 = tempBmm2Queue.AllocTensor<float>(); tmpBuffer3.SetSize(256);
父主题: LocalTensor