开发者
资源

Neg(ISASI)

产品支持情况

产品

是否支持

Atlas 350 加速卡

Atlas A3 训练系列产品/Atlas A3 推理系列产品

x

Atlas A2 训练系列产品/Atlas A2 推理系列产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品AI Core

x

Atlas 推理系列产品Vector Core

x

Atlas 训练系列产品

x

功能说明

按元素进行取相反数操作,计算公式如下:

函数原型

1
2
template <typename T>
__aicore__ inline void Neg(const LocalTensor<T>& dst, const LocalTensor<T>& src, const uint32_t count)

参数说明

表1 模板参数说明

参数名

描述

T

操作数数据类型。

Atlas 350 加速卡,支持的数据类型为:int8_t/int16_t/half/int32_t/float/int64_t

表2 参数说明

参数名

输入/输出

描述

dst

输出

目的操作数。

类型为LocalTensor,支持的TPosition为VECIN/VECCALC/VECOUT。

LocalTensor的起始地址需要32字节对齐。

src

输入

源操作数。

类型为LocalTensor,支持的TPosition为VECIN/VECCALC/VECOUT。

LocalTensor的起始地址需要32字节对齐。

源操作数的数据类型需要与目的操作数保持一致。

count

输入

参与计算的元素个数。

返回值说明

约束说明

调用示例

更多样例可参考LINK

1
AscendC::Neg(dstLocal, srcLocal, 512);
结果示例如下(srcLocal和dstLocal均为int64_t类型):
1
2
输入数据(srcLocal): [1 2 3 4 ...]
输出数据(dstLocal): [-1 -2 -3 -4 ...]