代码之家 › 专栏 › 技术社区 › user25004

如何将Pytroch张量从数学上的列向量转换为列矩阵?

tensor pytorch transpose vector matrix

user25004 · 技术社区 · 8 年前

我正在pytorch中使用张量。如何将对应于列向量的张量转换为对应于其转置的张量?

import numpy as np
coef = torch.from_numpy(np.arange(1.0, 5.0)).float()
print(coef)
print(coef.size())

当前大小为 coef 是 [4] 但我希望是这样 [4, 1] 内容相同。

2 回复 | 直到 8 年前

jdhao 8 年前

在Pytork中很容易实现。您可以使用 view() 方法

coef = coef.view(4, 1)
print(coef.size()) # now the shape will be [4, 1]

nemo 8 年前

使用时 .view 总的来说,这当然是一个很好的选择,为了完整起见,我想补充一点,还有 .unsqueeze() 在指定索引处添加额外维度的方法(与 .squeeze() 删除统一维度的方法):

>>> coef = coef.unsqueeze(-1) # add extra dimension at the end
>>> coef.shape
torch.Size([4, 1])

对于一般的换位,您可以使用 .t() 方法

推荐文章

Jingyang Wang · 一维numpy阵列的张量幂

2 年前

enryu · Pytorch增加尺寸

2 年前

mxcx · 将torch.all应用于除第一个维度外的所有维度

2 年前

Ihsan Ullah Khan · RuntimeError:维度数必须是sparse_dim(59697)+dense_dim

2 年前

xposure · Tensorflow:训练神经网络时损失没有改善

8 年前

user179156 · 用于向tensorflow模型提供tfrecords字节的java api

8 年前

user25004 · 如何将Pytroch张量从数学上的列向量转换为列矩阵?

8 年前

sashaostr · Keras/Tensorflow中不同时期的训练率如何变化

8 年前

Tom · 如何重塑keras中的张量?

8 年前

Tom · 如何将tensorflow中的dropout应用于多维张量?

8 年前