功能

固定维度的指定下标加上某个特定值。

定义

struct IndexAddParam {
    enum IndexType {
        INDEX_UNDEFINED = 0,
        INDEX_ADD,
    };
    IndexType indexType = INDEX_UNDEFINED;
    int64_t axis = 0;
};

成员

成员名称	描述
indexType	指定下标需要执行的操作类型。
axis	输入Tensor需加上updates更新值的轴。可为负数。值小于var的维度数。

输入

参数	维度	数据类型	格式	描述
var	[-1, …, -1]	float16	ND	输入tensor。
indices	[-1]	int32	ND	指定固定维度的指定下标。
updates	[-1, …, -1]	float16	ND	输入tensor累加值（更新值），维度数与var一致。
alpha	[1, -1, …, -1]	float16	ND	输入Tensor累加值的累加次数。

输出

参数	维度	数据类型	格式	描述
var	[-1,…,-1]	float16	ND	与输入Tensor为同一Tensor，内存地址相同。

IndexAddOperation

功能

定义

成员

输入

输出