tensor如何实现转置_pytorch tensor 变换

pytorch_bannar.png

import numpy as np

import torch

甜点

在学习神经网时，我们总是喜欢将时间花在如何通过用代码实现模型上，而往往对 tensor 在网络每一层的变化似乎不那么在意，可能觉得观察 tensor 形状比较简单和枯燥。

pytorch 改变 tensor 形状的 Api

view/reshape 改变形状

Squeeze/unsqueeze 增加维度/删减维度

transpose/permute 变换维度

Expand/repeat 维度扩展

高维 tensor

对于高纬 tensor，我们主要理解好后 2 个维度，可以理解为平面，3 维表示立体形状，随着维度增加我们就可以将每一个维度理解为容器或者盒子，更高维可以理解为装着低纬的容器或盒子。

改变形状

在 numpy 中使用 reshape 对 tensor 的形状进行改变，而在 pytorch 我们可以用 view 和 reshape 方法对 tensor 形状进行改变，他们除了名字不同，其他并没有什么区别，所以这里就以 view 为例来说一说如何改变 tensor 形状。