.unsqueeze(0)
有多余的1维轴,就会去掉
ref
https://stackoverflow.com/questions/42866654/what-is-the-difference-between-view-and-unsqueeze-in-torch
torch.unsqueeze() .view()
猜你喜欢
转载自blog.csdn.net/qq_35608277/article/details/85341902
今日推荐
周排行