import torch
x = torch.arange(18).view(3, 2, 3) x
tensor([[[ 0, 1, 2], [ 3, 4, 5]], [[ 6, 7, 8], [ 9, 10, 11]], [[12, 13, 14], [15, 16, 17]]])
x[1, 1, 1]
tensor(10)
x[1, 0:2, 1]
tensor([ 7, 10])
x[1, 0:2, 0:5]
tensor([[ 6, 7, 8], [ 9, 10, 11]])
x[1, :, :]