溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本篇內容主要講解“torch.argmin()怎么使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“torch.argmin()怎么使用”吧!
代碼實驗舉例:
C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python Python 3.7.7 (default, May 6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32 Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x000001FE6BDED330>>>>>>> a = torch.randn(4, 5)>>> a tensor([[ 0.5816, 2.0060, 1.6013, -0.6379, -0.4996],[ 0.2791, -0.5900, 0.2527, 0.2797, -0.5316],[ 0.6480, 2.6538, -0.2797, -1.3232, -0.8263],[-0.4343, -0.2370, -1.7016, -0.2565, 1.4568]])>>> torch.argmin(a)tensor(17)>>> torch.argmin(a, dim=1)tensor([3, 1, 3, 2])>>> torch.argmin(a, dim=0)tensor([3, 1, 3, 2, 2])>>>>>>>>>
到此，相信大家對“torch.argmin()怎么使用”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
