Pytorch：将张量的指数增加到特定的大小

Question

我有一个[20, 3, 32, 32]的张量 我想将3增加到64（[20,64,32,32]，其中20是批量大小）。我尝试了重复功能。但这只会给我63或66，因为使用repeat只能平铺（乘法）索引。
我如何解决这个问题？
谢谢

Answer 1

您可以重复使用^{}，然后使用索引进行过滤

t = torch.rand(20,3,32,32)
t.shape
torch.Size([20, 3, 32, 32])

t = t.repeat(1,22,1,1)[:,:-2,:,:]
t.shape
torch.Size([20, 64, 32, 32])