torch.squeeze()
压缩大小为1的维度
参考:https://pytorch.org/docs/stable/generated/torch.squeeze.html
torch.unsqueeze()
在指定的维度处增加一个维度，大小为1
参考：https://pytorch.org/docs/stable/generated/torch.unsqueeze.html
【d2l】【常见函数】【19】 torch.squeeze( ), torch.unsqueeze( )
发布时间 2023-07-29 18:02:14作者: zz子木zz