三维数组转四维

我是小小懒 2022-04-06 阅读 71

用reshape（）函数即可

    input = torch.randn(5, 7, 7)
    input=input.reshape(1,5, 7, 7)

0 条评论

关注