TileWithAxis

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：沿指定维度复制输入Tensor数据，扩展输出Tensor。

• 计算示例：

  • 输入 x = [[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12]]]，shape = (2, 3, 2)
  • axis = 1，tiles = 2
  • 输出 y = [[[1, 2], [3, 4], [5, 6], [1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12], [7, 8], [9, 10], [11, 12]]]，shape = (2, 6, 2)

• 计算公式： 输出 y 的 shape 与输入 x 相同，仅 axis 维度变为 x.shape[axis] * tiles。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量	FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64	ND
axis	属性（可选）	指定复制的维度，默认为1	INT	-
tiles	属性（必选）	复制次数，必须大于0	INT	-
y	输出	输出张量，与x具有相同的数据类型和格式	与x一致	ND

约束说明

• axis 必须在输入Tensor的维度范围内（支持负数索引，如 axis=-1 表示最后一维）。
• tiles 必须大于0。
• 输出Tensor在 axis 维度的大小必须等于输入Tensor在 axis 维度的大小乘以 tiles。
• 输入Tensor维度不超过8维。

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_tile_with_axis	通过算子IR构图方式调用TileWithAxis算子。