如下所示：

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大小例項就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。