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小时内删除。
发表评论