Когда я пытаюсь индексировать массив, я использую этот код для печати части столбца с помощью Numpy или Pytorch.
import numpy as np
a = np.random.randn(5,3)
a[:,1]
или
import torch
a = torch.Tensor(5,3)
a[:,1]
Вывод отображается следующим образом.
array([-0.07478094, -1.87787326, 0.50407517, 1.13335836, 0.23140931])
Но я хочу отобразить вывод в виде столбца. (Потому что я проиндексировал столбец)
array([-0.07478094,
-1.87787326,
0.50407517,
1.13335836,
0.23140931])
Кроме того, когда я создаю тензор с torch.ones (5), результат будет
tensor([1., 1., 1., 1., 1.])
но я хочу увидеть тип вывода на кнопке, как это
tensor([1., 1., 1., 1., 1.])
[torch.FloatTensor of size 5]
Причина, по которой я хочу отобразить это, заключается в том, что я не могу различить тензор и нюмпи
Может кто-нибудь сказать мне, как это сделать? Спасибо.