Pytorch矢量与空的尺寸

时间:2017-08-09 01:21:22

标签: python vector deep-learning pytorch

我注意到pytorch中的一些向量具有空a=torch.randn(1)的大小(1L,)。这些类型的向量被称为什么,为什么它们缺少第二维?我在OSX和Ubuntu上 - 不确定这是否有所作为。

示例:

>>> import torch
>>> a=torch.randn(1)
>>> a

 2.1358
[torch.FloatTensor of size 1]

>>> a.size()
(1L,)
>>> a=torch.randn(1).size()
>>> a
(1L,)

1 个答案:

答案 0 :(得分:0)

这不是特定的pytorch。这个约定也在python中遵循。设大小(5,)的矢量和大小(5,1)的矩阵是不同的。虽然它们具有相同数量的元素。

>>> import numpy as np
>>> a=np.random.rand(5)
>>> a.shape
>>> (5,)

>>> b=np.random.rand(5,1)
>>> (5,1)