1.torch.squeeze()对应numpy.squeeze()
2.torch.unsqueeze()对应numpy.expand_dims()
3.torch.permuate()对应numpy.transpose()
(慢慢积累,若有不对的地方,感谢指正,有用的话帮作者点个赞,感谢)
欢迎分享,转载请注明来源:内存溢出
1.torch.squeeze()对应numpy.squeeze()
2.torch.unsqueeze()对应numpy.expand_dims()
3.torch.permuate()对应numpy.transpose()
(慢慢积累,若有不对的地方,感谢指正,有用的话帮作者点个赞,感谢)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)