input = torch.randn(1, 3, 28, 36)
print(input.size())
print(input[..., 0].shape)
print(input[..., 1].shape)
print(input[..., 2].shape)
print(input[..., 3].shape)
print(input[..., 4].shape)
print(input[..., 5].shape)
結果:
torch.Size([1, 3, 28, 36])
torch.Size([1, 3, 28])
torch.Size([1, 3, 28])
torch.Size([1, 3, 28])
torch.Size([1, 3, 28])
torch.Size([1, 3, 28])
torch.Size([1, 3, 28])