Pytorch的常规操作
改变张量形状的操作 .view()
改变张量x的维度大小
x.view(-1, 1024)
-1
的作用是自动计算该维度的大小,以确保总元素数量不变。-1
的意思是 "任何可以使得总元素数量保持不变的值"。
1024
是目标形状中的另一个维度的大小。
x = x.view(-1, 3, 3)
-1自动计算x的维度大小 后面的两个三代表一个3+3矩阵
改变张量形状的操作 .view()
改变张量x的维度大小
x.view(-1, 1024)
-1
的作用是自动计算该维度的大小,以确保总元素数量不变。-1
的意思是 "任何可以使得总元素数量保持不变的值"。
1024
是目标形状中的另一个维度的大小。
x = x.view(-1, 3, 3)
-1自动计算x的维度大小 后面的两个三代表一个3+3矩阵