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

>>阅读原文<<