pytorch 笔记之squeeze
squeeeze使用
torch.squeeze(input, dim=None, out=None) → Tensor
Returns a tensor with all the dimensions of input of size 1 removed.
简单来说,squeeze只压缩所有长度为1的维度。如果指定dim(int, optional),则在指定的维度上压缩,且只...
原创
2018-09-07 22:23:19 ·
755 阅读 ·
0 评论