VB.net 2010 视频教程 VB.net 2010 视频教程 python基础视频教程
SQL Server 2008 视频教程 c#入门经典教程 Visual Basic从门到精通视频教程
当前位置:
首页 > temp > 简明python教程 >
  • 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。

 View Code

2.2 输出显示

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


相关教程