您好,登錄后才能下訂單哦!
本篇內容主要講解“Python torch.numel(input)怎么使用”,感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷,實用性強。下面就讓小編來帶大家學習“Python torch.numel(input)怎么使用”吧!
函數使用說明: torch.numel(input) → int 返回輸入張量input的元素總數
Microsoft Windows [版本 10.0.18363.1256](c) 2019 Microsoft Corporation。保留所有權利。 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 0x0000021CCA53D330>>>>>>> a = torch.randn(1, 2, 3, 4, 5)>>> a.numel()120>>> torch.numel(a)120>>>>>>>>> a = torch.zeros(4,4)>>> a.numel()16>>>>>>>>> torch.numel(a)16>>>>>>>>>
到此,相信大家對“Python torch.numel(input)怎么使用”有了更深的了解,不妨來實際操作一番吧!這里是億速云網站,更多相關內容可以進入相關頻道進行查詢,關注我們,繼續學習!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。