Use torch.max() along a dimension. However, you may wish to get the maximum along a particular dimension, as a Tensor, instead of a single element ...
確定! 回上一頁