脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → pytorch max eq

pytorch 常用函数 max ,eq说明

 更新时间：2020年06月28日 09:37:32   作者：蓝鲸123  
这篇文章主要介绍了pytorch 常用函数 max eq说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

max找出tensor 的行或者列最大的值：

找出每行的最大值:

import torch
outputs=torch.FloatTensor([[1],[2],[3]])
print(torch.max(outputs.data,1))

输出：

(tensor([ 1., 2., 3.]), tensor([ 0, 0, 0]))

找出每列的最大值：

import torch
outputs=torch.FloatTensor([[1],[2],[3]])
print(torch.max(outputs.data,0))

输出结果：

(tensor([ 3.]), tensor([ 2]))

Tensor比较eq相等：

import torch

outputs=torch.FloatTensor([[1],[2],[3]])
targets=torch.FloatTensor([[0],[2],[3]])
print(targets.eq(outputs.data))

输出结果：

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

使用sum() 统计相等的个数：

import torch

outputs=torch.FloatTensor([[1],[2],[3]])
targets=torch.FloatTensor([[0],[2],[3]])
print(targets.eq(outputs.data).cpu().sum())

输出结果：

tensor(2)

补充知识：PyTorch - torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le

flyfish

torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le

以上全是简写

参数是input, other, out=None

逐元素比较input和other

返回是torch.BoolTensor

import torch

a=torch.tensor([[1, 2], [3, 4]])
b=torch.tensor([[1, 2], [4, 3]])

print(torch.eq(a,b))#equals
# tensor([[ True, True],
#     [False, False]])

print(torch.ne(a,b))#not equal to
# tensor([[False, False],
#     [ True, True]])

print(torch.gt(a,b))#greater than
# tensor([[False, False],
#     [False, True]])

print(torch.lt(a,b))#less than
# tensor([[False, False],
#     [ True, False]])

print(torch.ge(a,b))#greater than or equal to
# tensor([[ True, True],
#     [False, True]])

print(torch.le(a,b))#less than or equal to
# tensor([[ True, True],
#     [ True, False]])

以上这篇pytorch 常用函数 max ,eq说明就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python适配器模式代码实现解析

    Python适配器模式代码实现解析

    这篇文章主要介绍了Python适配器模式代码实现解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • 基于python 将列表作为参数传入函数时的测试与理解

    基于python 将列表作为参数传入函数时的测试与理解

    这篇文章主要介绍了基于python 将列表作为参数传入函数时的测试与理解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • python使用ctypes库调用DLL动态链接库

    python使用ctypes库调用DLL动态链接库

    这篇文章主要介绍了python如何使用ctypes库调用DLL动态链接库，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
    2020-10-10
  • 利用matlab与Excel交互之单元格操作

    利用matlab与Excel交互之单元格操作

    Excel是广泛使用的“电子表格”,Matlab则具有强大的数值计算、统计分析以及图形可视化能力,这篇文章主要给大家介绍了关于利用matlab与Excel交互之单元格操作的相关资料,需要的朋友可以参考下
    2021-08-08
  • Python3解释器知识点总结

    Python3解释器知识点总结

    在本篇内容中小编给大家总结了关于Python3解释器的用法以及相关知识点，需要的朋友们学习下。
    2019-02-02
  • OpenCV之理解KNN邻近算法k-Nearest Neighbour

    OpenCV之理解KNN邻近算法k-Nearest Neighbour

    这篇文章主要为大家介绍了OpenCV之理解KNN邻近算法k-Nearest Neighbour，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2023-05-05
  • Python实现按特定格式对文件进行读写的方法示例

    Python实现按特定格式对文件进行读写的方法示例

    这篇文章主要介绍了Python实现按特定格式对文件进行读写的方法,可实现文件按原有格式读取与写入的功能,涉及文件的读取、遍历、转换、写入等相关操作技巧,需要的朋友可以参考下
    2017-11-11
  • Python中内置数据类型list,tuple,dict,set的区别和用法

    Python中内置数据类型list,tuple,dict,set的区别和用法

    这篇文章主要给大家介绍了Python中内置数据类型list,tuple,dict,set的区别和用法，都是非常基础的知识，十分的细致全面，有需要的小伙伴可以参考下。
    2015-12-12
  • Python中函数的返回值示例浅析

    Python中函数的返回值示例浅析

    这篇文章主要给大家介绍了关于Python中函数返回值的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
    2019-08-08
  • 在 Python 中接管键盘中断信号的实现方法

    在 Python 中接管键盘中断信号的实现方法

    要使用信号，我们需用导入 Python 的signal库。然后自定义一个信号回调函数，当 Python 收到某个信号时，调用这个函数。 ，下面通过实例代码给大家介绍在 Python 中接管键盘中断信号,需要的朋友可以参考下
    2020-02-02

最新评论