pytorch相关总结

以下全部是实例，通过例子更能体会相关函数的作用。 一、view()和expand() x=torch.Tensor([[1,2],[3,4]]) H=x.shape[0] W=x.shape[1] x1 = x.view(H, 1, W, 1) x1 输出： tensor([[[[1.], [2.]]], [[[3.], [4.]]]]) x

>>阅读原文<<