Pytorch学习之torch用法—-比较操作(ComparisonOps)资源-CSDN文库

175 浏览量 2020-12-17 07:31:00 上传评论收藏 55KB PDF 举报

资源推荐

资源详情

资源评论

Pytorch学习之学习之torch用法用法—-比较操作比较操作(Comparison Ops)

1. torch.eq(input, other, out=None)

说明：比较元素是否相等，第二个参数可以是一个数，或者是第一个参数同类型形状的张量

参数：

input(Tensor) —- 待比较张量

other(Tenosr or float) —- 比较张量或者数

out(Tensor,可选的) —- 输出张量

返回值：一个torch.ByteTensor张量，包含了每个位置的比较结果(相等为1，不等为0)

>>> a = torch.Tensor([[1, 2], [3, 4]])

>>> b = torch.Tensor([[1, 1], [4, 4]])

>>> torch.eq(a, b)

tensor([[1, 0],

[0, 1]], dtype=torch.uint8)

2. torch.equal(tensor1, tensor2, out=None)

说明：如果两个张量有相同的形状和元素值，则返回true，否则False

参数：

tensor1(Tenosr) —- 比较张量1

tensor2(Tensor) —- 比较张量2

out(Tensor,可选的) —- 输出张量

>>> a = torch.Tensor([1, 2])

>>> b = torch.Tensor([1, 2])

>>> torch.equal(a, b)

True

3. torch.ge(input, other, out=None)

说明：逐元素比较input和other，即是否input >= other。

参数：

input(Tensor) —- 待对比的张量

other(Tensor or float) —- 对比的张量或float值

out(Tensor,可选的) —- 输出张量，

>>> a = torch.Tensor([[1, 2], [3, 4]])

>>> b = torch.Tensor([[1, 1], [4, 4]])

>>> torch.ge(a, b)

tensor([[1, 1],

[0, 1]], dtype=torch.uint8)

4. torch.gt(input, other, out=None)

说明：逐元素比较input和other，即是否input > other

参数：

input(Tensor) —- 要对比的张量

other(Tensor or float) —- 要对比的张量或float值

out(Tensor,可选的) —- 输出张量

>>> a = torch.Tensor([[1, 2], [3, 4]])

>>> b = torch.Tensor([[1, 1], [4, 4]])

本内容试读结束，登录后可阅读更多

下载后可阅读完整内容，剩余4页未读，立即下载

内容反馈

付出余切

粉丝: 200
资源: 912

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈

feedback-tip