Pytorch 中的view函数作用是将张量铺平,
例如:
import torch target = torch.randn(10,2) print(target) target = target.view(1,-1) print(target)
tensor([[ 0.4080, 0.2592],
[-0.6065, -1.6695],
[-0.2938, 1.0712],
[-1.2811, 0.1640],
[-0.1256, 0.3585],
[-0.6682, -0.8516],
[-1.6582, 1.2890],
[ 0.0135, 1.5360],
[-0.1070, 0.6072],
[-0.1969, -0.6077]])
tensor([[ 0.4080, 0.2592, -0.6065, -1.6695, -0.2938, 1.0712, -1.2811, 0.1640,
-0.1256, 0.3585, -0.6682, -0.8516, -1.6582, 1.2890, 0.0135, 1.5360,
-0.1070, 0.6072, -0.1969, -0.6077]])

本文介绍了PyTorch框架中view函数的功能及其使用方法。通过一个实例展示了如何利用view函数来改变张量的形状,将多维张量转换为一维张量,方便进行后续的处理工作。
1570

被折叠的 条评论
为什么被折叠?



