tensor的复制函数torch.repeat_interleave()

tensor的复制函数torch.repeat_interleave()

1. repeat_interleave(self: Tensor, repeats: _int, dim: Optional[_int]=None)

参数说明：

self: 传入的数据为tensor

repeats: 复制的份数

dim: 要复制的维度，可设定为0/1/2.....

2. 例子

2.1 Code

此处定义了一个4维tensor，要对第2个维度复制，由原来的1变为3，即将设定dim=1。

2.2 输出显示

即可看到输入tensor形状为[2, 1, 3, 3]，经过repeat后，tensor变为[2, 3, 3, 3]，并在第二维度上保持相同的数据。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。