tensor([[[2., 1., 2., 2., 3., 1., 3., 3., 3., 1.],
[3., 2., 2., 3., 3., 1., 2., 3., 3., 2.],
[1., 1., 3., 2., 3., 2., 1., 1., 2., 2.],
[3., 2., 3., 2., 3., 1., 1., 3., 1., 1.],
[2., 3., 3., 2., 1., 2., 3., 1., 2., 3.],
[1., 1., 2., 3., 3., 1., 1., 2., 2., 1.],
[3., 2., 1., 3., 3., 1., 3., 1., 3., 2.],
[3., 3., 2., 2., 3., 1., 2., 2., 1., 2.],
[1., 1., 3., 1., 3., 3., 1., 2., 3., 1.],
[3., 2., 2., 1., 1., 1., 1., 1., 2., 2.]]], device='cuda:0',
grad_fn=<SelectBackward>)
tensor([[[2., 2., 1., 1., 3., 1., 1., 1., 1., 3.],
[2., 1., 3., 3., 2., 2., 1., 1., 2., 3.],
[2., 2., 2., 2., 1., 2., 2., 1., 1., 2.],
[1., 3., 3., 3., 3., 3., 1., 1., 1., 3.],
[1., 3., 1., 2., 3., 1., 2., 3., 3., 1.],
[3., 1., 3., 1., 3., 2., 1., 2., 1., 2.],
[3., 2., 2., 1., 1., 2., 2., 1., 2., 1.],
[3., 2., 2., 1., 1., 3., 2., 1., 1., 1.],
[1., 2., 3., 1., 1., 3., 3., 3., 3., 1.],
[3., 2., 1., 3., 3., 3., 2., 2., 2., 2.]]], device='cuda:0',
grad_fn=<SelectBackward>)
torch.Size([1, 3, 3, 10, 10])
tensor([[0., 0., 0.],
[0., 4., 4.],
[0., 4., 2.]], device='cuda:0', grad_fn=<SelectBackward>)