将输入Tensor的Shape，按指定轴扩展指定的倍数。
输出y的维度和multiples维度一致，每个维度大小为输入x广播到multiples维度后和multiples对应维度的乘积。
struct RepeatParam {
SVector<int64_t> multiples;
};
成员名称 |
描述 |
|---|---|
multiples |
每一维度上扩展的倍数。 支持在不超过两个维度上进行扩展。 multiples的维度小于等于8且需大于或等于输入x的维度，每一个元素要求大于0。 |
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |
参数 |
维度 |
数据类型 |
格式 |
|---|---|---|---|
y |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16 |
ND |