Hi guys,
I’m new to pytorch. I wonder if pytorch support to indexing specified dimensions for example, for a tensor a:
a = torch.FloatTensor([[1, 2, 3], [4, 5, 6]])
I want to select the first and third column as a subtensor.
[[1, 3]
[4, 6]]
I checked documents and discussion but still don’t know how to do it. I need some guidance.
Thanks!