Pytorch: увеличить индекс тензора до заданного размера c

Question

У меня есть тензор [20, 3, 32, 32] Я хочу увеличить 3 до 64 ([20,64,32,32], где 20 - размер партии). Я попробовал повторить функцию. Но это только дает мне 63 или 66, потому что с повторением вы можете только умножать (умножать) индексы.
Как я могу решить это?
Спасибо!

Answer 1

Вы можете повторить, используя torch.repeat, а затем отфильтровать по индексам.

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])