評價此頁

torch.Tensor.argmax#

Tensor.argmax(dim=None, keepdim=False) LongTensor#

請參閱 torch.argmax()