使用equal 张量比较 【python|Pytorch中判断两个tensor是否相等】原型:equal(other)
比较两个张量是否相等–相等返回:True;
否则返回:False
import torch
a = torch.tensor([4, 2, 3, 4])
b = torch.tensor([4, 2, 3, 4])print(a.equal(b)# 返回结果:True
使用eq 逐元素判断
原型:eq(other)
比较两个张量tensor中,每一个对应位置上元素是否相等–对应位置相等,就返回一个1;否则返回一个0.
import torcha = torch.tensor([2, 2, 2, 3])
b = torch.tensor([2, 2, 3, 3])print(a.eq(b))# 返回结果:tensor([True, True, True, True])
使用eq_将判断结果返回并替换原tensor 原型:eq_(other)
等价于tensor = tensor.eq(other);
即:将比较后的结果替换原张量的值
import torcha = torch.tensor([1, 2, 3])
b = torch.tensor([1, 1, 3])print(a.eq_(b))
print(a)# 返回结果:tensor([1, 0, 1])
tensor([1, 0, 1])# 可以看到a中的值已经被替换了
推荐阅读
- 日常小技巧|任意角度旋转图片(python)
- 日常小技巧|已知点坐标,求夹角大小(0-360度)(python)
- 笔记|Day30.守护线程、定时器、消费者模式 | wait、notify方法
- Python|用Python去除扫描型PDF中的水印
- Python数据可视化|Python数据可视化之随机漫步
- 自学python|数据可视化_模拟随机漫步
- python|numpy刷题——02
- 钉钉自动打卡
- 博图PLC程序 停车场控制系统