Pytorch的常规操作

改变张量形状的操作   .view()

改变张量x的维度大小

x.view(-1, 1024)

-1 的作用是自动计算该维度的大小,以确保总元素数量不变。-1 的意思是 "任何可以使得总元素数量保持不变的值"。

1024 是目标形状中的另一个维度的大小。

x = x.view(-1, 3, 3)

-1自动计算x的维度大小 后面的两个三代表一个3+3矩阵

你可能感兴趣的:(计算机视觉科研笔记,pytorch,人工智能,python)