在维度中使用 None 可以在所在维度中增加一维。
2、使用 unsqueeze() 增加 tensor 的维度torch.unsqueeze(input, dim) → Tensor
torch.Tensor.unsqueeze(dim) → Tensor
torch.squeeze(input, dim=None) → Tensor
torch.Tensor.squeeze(dim=None) → Tensor
在维度中使用 None 可以在所在维度中增加一维。
2、使用 unsqueeze() 增加 tensor 的维度torch.unsqueeze(input, dim) → Tensor
torch.Tensor.unsqueeze(dim) → Tensor
torch.squeeze(input, dim=None) → Tensor
torch.Tensor.squeeze(dim=None) → Tensor
版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。