0
点赞
收藏
分享

微信扫一扫

pytorch的tf.transpose

干自闭 2022-07-27 阅读 85


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])


举报

相关推荐

0 条评论