pytorch逐元素⽐较tensor⼤⼩实例
如下所⽰：
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([[Fal, True],
[Fal, True]])
我们取出tenor a中对应的⼤于0.01的值：黄油饼干的做法
再生混凝土
开机启动设置a[mask]
将对应满⾜条件的元素输出并⾃动拉伸为⼀个⼀维向量输出：公司入股
tensor([0.0110, 0.9000])
我们也可以⽐较两个tensor⼤⼩
责任的英语
b = sor([[0.02, 1], [0, 1.0]])
<(a, b)
tensor([[Fal, Fal],
赛龙舟作文[ True, Fal]])
大叫的英文以上这篇pytorch逐元素⽐较tensor⼤⼩实例就是⼩编分享给⼤家的全部内容了，希望能给⼤家⼀个参考，也希望⼤家多多⽀持。
