maxframe.tensor.repeat

maxframe.tensor.repeat(a, repeats, axis=None)

重复张量的元素。

参数:

• a (array_like) -- 输入张量。

• repeats (int or tensor of ints) -- 每个元素的重复次数。repeats 会被广播以适应给定轴的形状。

• axis (int, optional) -- 沿其重复值的轴。默认情况下，使用展平的输入张量，并返回一个展平的输出张量。

返回:

repeated_tensor -- 输出数组，其形状与 a 相同，除了沿给定轴外。

返回类型:

Tensor

参见

tile

平铺一个张量。

示例

>>> import maxframe.tensor as mt

>>> mt.repeat(3, 4).execute()
array([3, 3, 3, 3])
>>> x = mt.array([[1,2],[3,4]])
>>> mt.repeat(x, 2).execute()
array([1, 1, 2, 2, 3, 3, 4, 4])
>>> mt.repeat(x, 3, axis=1).execute()
array([[1, 1, 1, 2, 2, 2],
       [3, 3, 3, 4, 4, 4]])
>>> mt.repeat(x, [1, 2], axis=0).execute()
array([[1, 2],
       [3, 4],
       [3, 4]])