import torch
ls2 = torch.Tensor([0.2,0.2,0.1])
print(ls2.eq(0.2))
print(ls2.eq(0.2).float()) 
代码理解 eq
于 2024-03-09 21:25:59 首次发布
文章展示了如何在PyTorch中使用`torch.Tensor`进行数值比较（如检查是否等于特定值）以及类型转换（将布尔张量转换为浮点数）。
368
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
