However, once I started to play around with 2D and 3D tensors and to sum over rows and columns, I got confused mostly about the second parameter dim of torch.
確定! 回上一頁