【d2l】【常见函数】【19】 torch.squeeze( ), torch.unsqueeze( )

torch.squeeze()

压缩大小为1的维度
参考:https://pytorch.org/docs/stable/generated/torch.squeeze.html

torch.unsqueeze()

在指定的维度处增加一个维度，大小为1
参考：https://pytorch.org/docs/stable/generated/torch.unsqueeze.html