本篇內容主要講解“torch.argmin()怎么使用”,感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷,實用性強。下面就讓小編來帶大家學習“torch.argmin()怎么使用”吧!
創(chuàng)新互聯(lián)專業(yè)為企業(yè)提供青河網(wǎng)站建設、青河做網(wǎng)站、青河網(wǎng)站設計、青河網(wǎng)站制作等企業(yè)網(wǎng)站建設、網(wǎng)頁設計與制作、青河企業(yè)網(wǎng)站模板建站服務,十多年青河做網(wǎng)站經(jīng)驗,不只是建網(wǎng)站,更提供有價值的思路和整體網(wǎng)絡服務。
代碼實驗舉例:
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)>>>>>> 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()怎么使用”有了更深的了解,不妨來實際操作一番吧!這里是創(chuàng)新互聯(lián)網(wǎng)站,更多相關內容可以進入相關頻道進行查詢,關注我們,繼續(xù)學習!