捷徑

torch.Tensor.argmax

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

請參閱 torch.argmax()

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得為初學者和進階開發者設計的深入教學

檢視教學

資源

尋找開發資源並取得問題解答

檢視資源