tensor的复制函数torch.repeat_interleave()
程序员文章站
2022-05-04 08:56:29
1. repeat_interleave(self: Tensor, repeats: _int, dim: Optional[_int]=None) 参数说明: self: 传入的数据为tensor repeats: 复制的份数 dim: 要复制的维度,可设定为0/1/2..... 2. 例子 2 ......
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。
1 import torch 2 3 4 def function(): 5 data1 = torch.rand([2, 1, 3, 3]) 6 print("data1_shape: ", data1.shape) 7 print("data1: ", data1) 8 9 data2 = torch.repeat_interleave(data1, repeats=3, dim=1) 10 print("data2_shape: ", data2.shape) 11 print("data2: ", data2) 12 13 14 if __name__ == '__main__': 15 function()
2.2 输出显示
即可看到输入tensor形状为[2, 1, 3, 3],经过repeat后,tensor变为[2, 3, 3, 3],并在第二维度上保持相同的数据。
推荐阅读
-
PHP操作文件类的函数代码(文件和文件夹创建,复制,移动和删除)
-
PHP递归复制、移动目录的自定义函数分享
-
C++类的相关问题、构造函数与析构函数、复制构造函数实例讲解
-
tensor的复制函数torch.repeat_interleave()
-
实现一个函数clone,可以对JavaScript中的5种主要的数据类型(包括Number、String、Object、Array、Boolean)进行值复制
-
[转]为什么复制构造函数的参数需要加const和引用
-
复制构造函数传递的是引用而非指针
-
php递归函数递归复制与删除目录的那些事
-
MySQL复制表结构和表数据的SQL语句和时间函数
-
关于该封闭类的复制构造函数问题