a = torch.randn(3,2)
for example:
tensor([[ 0.5957, 1.6250],
[-0.4354, 2.0719],
[ 0.8076, -1.2720]])
and a[1] = tensor([-0.4354, 2.0719])
, but I want to get the cols’ data of a,
such as tensor([1.6250,2.0719,-1.2720])
, what shoud I do? thanks for your help!