pytorch逐元素比較tensor大小

2021-09-26 05:28:52 字數 614 閱讀 8834

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]])

Tensor 逐元素操作

逐元素操作 這部分操作會對tensor的每乙個元素 point wise,又名element wise 進行操作,此類操作的輸入與輸出形狀一致。常用的操作如表3 4所示。表3 4 常見的逐元素操作 函式功能 abs sqrt div exp fmod log pow.絕對值 平方根 除法 指數 求餘...

C 的string逐位處理效率比較

今天看到乙個對string做逐位處理的 我看見 使用指標來指向string,我就在想為什麼不用中括號,型如string s s i 這種操作。想到這個 作者一向以效率優先考慮,於是我測試了一下效率,果不其然。哈哈 include include include include include inc...

通用函式 快速的逐元素陣列函式

通用函式也可以稱為 ufunc,是一種在 ndarray 資料中進行逐元素操作的函式。某些簡單函式接受了乙個或者多個標量數值,並產生乙個或多個標量結果,而通用函式就是對這些簡單函式的向量化封裝。有很多 ufunc 是簡單的逐元素轉換,比如 sqrt 和 exp 函式 import numpy as ...