Pytorch中常用的一些函数
文章目录维度相关
维度相关
torch.squeeze(input, dim=None, out=None) → Tensor
Tensor.squeeze(dim=None) → Tensor
返回一个tensor,这个tensor是将input的所有维度大小为1维度去掉得到的
若不指定dim,则去掉所有维度为1的维度,若指定维度,且该维度为1,则去掉这个维度
torch.unsque...
原创
2020-03-07 17:13:02 ·
1004 阅读 ·
0 评论