pytorch的tf.transpose

来源:互联网 发布:数据精灵稳定版源 编辑:程序博客网 时间:2024/06/14 18:49

permute

>>> img_nhwc = torch.randn(10, 480, 640, 3)>>> img_nhwc.size()torch.Size([10, 480, 640, 3])>>> img_nchw = img_nhwc.permute(0, 3, 1, 2)>>> img_nchw.size()torch.Size([10, 3, 480, 640])
原创粉丝点击