torch.tensor.repeat
时间: 2023-11-03 22:52:28 浏览: 63
torch练习.py
torch.tensor.repeat()函数可以对张量进行重复扩充。当参数只有两个时,表示行的重复倍数和列的重复倍数,1表示不重复。当参数有三个时,表示通道数的重复倍数、行的重复倍数和列的重复倍数,1表示不重复。举个例子,如果输入一个一维张量,参数为一个,即表示在列上进行重复n次。例如,使用a = torch.randn(3)创建一个一维张量a,然后使用a.repeat(4)进行重复扩充,结果会将a重复四次,形成一个新的张量。输出结果为(tensor([ 0.81, -0.57, 0.10]), tensor([ 0.81, -0.57, 0.10, 0.81, -0.57, 0.10, 0.81, -0.57, 0.10, 0.81, -0.57, 0.10]))。
阅读全文