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

我有一个张量

[20,3,32,32]

我想将

3

增加到

64

（

[20,64,32,32]

，其中

20

是批量大小）。我尝试了重复功能。但这只能给我

63

或

66

，因为重复只能平铺（乘以）索引。
我如何解决这个问题？
谢谢

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

t=torch.rand（20,3,32,32）
t、 形状
火炬尺寸（[20,3,32,32]）
t=t.重复（1,22,1,1）[：，：-2，：，：]
t、 形状
火炬尺寸（[20,64,32,32]）

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

t=torch.rand（20,3,32,32）
t、 形状
火炬尺寸（[20,3,32,32]）
t=t.重复（1,22,1,1）[：，：-2，：，：]
t、 形状
火炬尺寸（[20,64,32,32]）