torch.unsqueeze() .view()

.unsqueeze(0)
有多余的1维轴,就会去掉
ref
https://stackoverflow.com/questions/42866654/what-is-the-difference-between-view-and-unsqueeze-in-torch

猜你喜欢

转载自blog.csdn.net/qq_35608277/article/details/85341902