产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
为Tensor添加用户自定义信息，用户可以根据需要设置对应的Tag。后续可通过GetUserTag获取指定Tensor的Tag信息，并根据Tag信息对Tensor进行相应操作。
1 | __aicore__ inline void SetUserTag(const TTagType tag) |
参数名 |
输入/输出 |
描述 |
|---|---|---|
tag |
输入 |
设置的Tag信息，类型TTagType对应为int32_t。 |
无
无
// 示例 AscendC::TTagType tag = 10; inputLocal.SetUserTag(tag); // 对LocalTensor设置tag信息。