squeeze是压缩张量的命令
import torch
a = torch.randn(1,3)
print(a)
print(a.shape)
比如说squeeze(?)括号里是啥 就是在哪个维度上删除维度为1 之后的结果
比如上上面那个里子 a是([[]])
但是在下面那个例子中d就是把0维上的维度为1的删除掉了 所以就是([])了
squeeze是压缩张量的命令
import torch
a = torch.randn(1,3)
print(a)
print(a.shape)
比如说squeeze(?)括号里是啥 就是在哪个维度上删除维度为1 之后的结果
比如上上面那个里子 a是([[]])
但是在下面那个例子中d就是把0维上的维度为1的删除掉了 所以就是([])了