我正在 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]
具有相同的内容。
在 PyTorch 中很容易实现。您可以使用该view()
方法。
coef = coef.view(4, 1)
print(coef.size()) # now the shape will be [4, 1]
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句