import torch
a = torch.tensor([[0.01, 0.011], [0.009, 0.9]])
mask = a.gt(0.01)
print(mask)
import torch
a = torch.tensor([[0.01, 0.011], [
阅读背景：
pytorch逐元素比较tensor大小_MD2017的博客_pytorch 逐元素比较
来源：互联网
分享到：
