pytorch逐元素⽐较tensor⼤⼩实例
trunc函数还能比大小吗
如下所⽰：
import torch
a = sor([[0.01, 0.011], [0.009, 0.9]])
mask = a.gt(0.01)
print(mask)
tensor⽐较⼤⼩可以⽤属性。上⾯⽐较了a中每个元素和0.01的⼤⼩，⼤于0.01的元素输出True。输出结果：tensor([[False, True],
[False, True]])
我们取出tenor a中对应的⼤于0.01的值：
a[mask]
将对应满⾜条件的元素输出并⾃动拉伸为⼀个⼀维向量输出：
tensor([0.0110, 0.9000])
我们也可以⽐较两个tensor⼤⼩
b = sor([[0.02, 1], [0, 1.0]])
<(a, b)
tensor([[False, False],
[ True, False]])
以上这篇pytorch逐元素⽐较tensor⼤⼩实例就是⼩编分享给⼤家的全部内容了，希望能给⼤家⼀个参考，也希望⼤家多多⽀持。

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。