Indeed, this gives us the global maximum element in the Tensor! Use torch.max() along a dimension. However, you may wish to get the maximum ...
確定! 回上一頁