torch.argmax. torch. argmax (input) → LongTensor. Returns the indices of the maximum value of all elements in the input tensor.
確定! 回上一頁