Python Jupyter Notebook如何在垂直方向显示矩阵

Question

当我尝试索引数组时，我使用此代码通过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]

我要显示此内容的原因是我无法区分张量和numpy

谁能告诉我该怎么做？谢谢。

Answer 1

尝试一下：

np.vstack(a)

希望这会有所帮助。