view
>>> import torch
>>> a = torch.randn(10, 480, 640, 3)
>>> b=a.view([480,10,640,3])
>>> b.size()
(480L, 10L, 640L, 3L)
pytorch的tf.reshape
阅读 29
2022-07-27
view
>>> import torch
>>> a = torch.randn(10, 480, 640, 3)
>>> b=a.view([480,10,640,3])
>>> b.size()
(480L, 10L, 640L, 3L)
相关推荐
精彩评论(0)