Как изменить изображение в pytorch

Question

У меня есть изображение формы (32, 3, 32, 32). Я знаю, что он имеет форму (batch_size, Channel, Height, Width).

Q. Как мне преобразовать его в (32, 32, 32), переопределяя Channel?

Answer 1

Если вы хотите преобразовать в оттенки серого, вы можете сделать это:

image.mean(dim=1)