IndexAddOperation

功能

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

定义

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

成员

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

成员名称

描述

IndexType

指定下标需要执行的操作类型。

INDEX_UNDEFINED：默认值。
INDEX_ADD：加。

indexType

指定下标需要执行的操作类型。

axis

输入Tensor需加上updates更新值的轴。可为负数。值小于var的维度数。

输入

参数	维度	数据类型	格式	描述
var	[-1, …, -1] -1表示当前维度的大小没有约束。	float16	ND	输入Tensor。
indices	[-1] -1表示当前维度的大小没有约束。	int32	ND	指定固定维度的指定下标。
updates	[-1, …, -1] -1表示当前维度的大小没有约束。	float16	ND	输入Tensor累加值（更新值）。
alpha	[1]	float16	ND	输入Tensor累加值的累加次数。

输出

参数	维度	数据类型	格式	描述
output	[-1,…,-1] -1表示当前维度的大小没有约束。	float16	ND	输出tensor。

参数

维度

数据类型

格式

描述

output

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16

输出tensor。

父主题： atb/infer_op_params.h