溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章主要介紹了pytorch逐元素比較tensor大小的示例分析，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。
如下所示：
import torch a = torch.tensor([[0.01, 0.011], [0.009, 0.9]]) mask = a.gt(0.01) print(mask)
tensor比較大小可以用tensor.gt屬性。上面比較了a中每個元素和0.01的大小，大于0.01的元素輸出True。輸出結果：
tensor([[False, True], [False, True]])
我們取出tenor a中對應的大于0.01的值：
a[mask]
將對應滿足條件的元素輸出并自動拉伸為一個一維向量輸出：
tensor([0.0110, 0.9000])
我們也可以比較兩個tensor大小
b = torch.tensor([[0.02, 1], [0, 1.0]]) torch.gt(a, b) tensor([[False, False], [ True, False]])
感謝你能夠認真閱讀完這篇文章，希望小編分享的“pytorch逐元素比較tensor大小的示例分析”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
