RepeatOperation

功能

将输入tensor的Shape,根据参数中multiples上的信息扩展指定倍数。

硬件支持情况

硬件型号

支持情况

特殊说明

Atlas 200I/500 A2 推理产品

支持

不支持输入输出Tensor数据类型为bf16的场景。

Atlas 推理系列产品

支持

不支持输入输出tensor数据类型为bf16的场景。

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

支持

-

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

支持

-

定义

1
2
3
4
struct RepeatParam {
    SVector<int64_t> multiples;
    uint8_t rsv[8] = {0};
};

参数列表

成员名称

类型

默认值

描述

multiples

SVector<int64_t>

-

输入tensor每个维度上扩展的倍数,支持对输入tensor的广播。

支持在不超过两个维度上进行扩展。

multiples的维度应小于等于8,且大于或等于输入tensor的维度,每一个元素要求都大于0。

rsv[8]

uint8_t

{0}

预留参数。

输入

参数

维度

数据类型

格式

描述

x

[x_dim_0,x_dim_1,... ,x_dim_n]

float16/bf16

ND

输入tensor。

输出

参数

维度

数据类型

格式

描述

y

[y_dim_0,y_dim_1,... ,y_dim_n]

float16/bf16

ND

输出tensor。和输入的数据类型一致。

规格约束

功能列表