torch.repeat()的用法

这两天写程序中需要，查到了一个非常好用的函数，烂笔头比好记性好用，记录下：

import torch
x = torch.tensor([1, 2, 3])
print(x.repeat(4))
print("###################################")
print(x.repeat(4, 1))

   
   

   
   
1
2
3
4
5

运行结果如下：

import torch
x = torch.tensor([1, 2, 3])
print(x.repeat(4, 1))
print("###################################")
print(x.repeat(4, 2))

   
   

   
   
1
2
3
4
5

运行结果如下：

import torch
x = torch.tensor([1, 2, 3])
print(x.repeat(4, 1))
print("###################################")
print(x.repeat(4, 2, 1))

   
   

   
   
1
2
3
4
5

运行结果如下：