0
点赞
收藏
分享

微信扫一扫

pytorch的tf.reshape

半夜放水 2022-07-27 阅读 29


view

>>> import torch
>>> a = torch.randn(10, 480, 640, 3)
>>> b=a.view([480,10,640,3])
>>> b.size()

(480L, 10L, 640L, 3L)


举报

相关推荐

0 条评论