【发布时间】:2018-09-15 09:20:44
【问题描述】:
当我尝试索引数组时,我使用此代码使用 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]
之所以要显示这个是因为我无法区分tensor和numpy
谁能告诉我怎么做?谢谢。
【问题讨论】:
-
尝试显示
a[:, [1]]。然后形状应该是 (n,1) 并显示为一列。 -
嘿@justin_sakong 试试下面的答案
标签: python-3.x numpy matrix pytorch display