pytorch的unsqueeze和unsqueeze_的使用
pytorch的unsqueeze和unsqueeze_的使用,及in_place操作unsqueezeunsqueeze就是用来扩展维度的, 先来看下unsqueeze的参数, torch.unsqueeze(input, dim, out=None), 扩维的范围是[-input.dim() - 1, input.dim() + 1), 如果是传入的dim是负数,那么dim实际数值是dim...
原创
2018-12-19 17:04:22 ·
12945 阅读 ·
0 评论