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

lycnight / 2023-07-29 / 原文

torch.squeeze()

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

torch.unsqueeze()

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