pytorch中tensor的squeeze()和unsqueeze()

本文转载自查看原文 2018-06-26 22:33 11398 pytorch

squeeze()：

squeeze(arg)表示第arg维的维度值为1，则去掉该维度。否则tensor不变。（即若tensor.shape()[arg] = 1，则去掉该维度）

unsqueeze()：

unsqueeze(arg)与squeeze(arg)作用相反，表示在第arg维增加一个维度值为1的维度。

常用操作：

参考：https://blog.csdn.net/lanse_zhicheng/article/details/79148678

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 Pytorch中的squeeze()和unsqueeze()函数 [ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法 pytorch 中改变tensor维度(transpose)、拼接(cat)、压缩(squeeze)详解 pytorch中的size()、 squeeze()函数 torch.squeeze()和torch.unsqueeze() pytorch 中tensor的加减和mul、matmul、bmm [Pytorch]Pytorch中tensor常用语法 pytorch 中Tensor.uniform_代替numpy.random.uniform pytorch中的标准化的tensor转图像 Pytorch中的torch.cat()函数 tensor拼接