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使用keys() 获取 Redis 数据库中的所有键
哈希、列表、集合、排序集合、字符串、JSON 和流是 Redis 支持的众多数据结构之一,本文将讨论获取 Redis 数据库中的所有键,感兴趣的朋友跟随小编一起看看吧2023-08-08浅谈tensorflow使用张量时的一些注意点tf.concat,tf.reshape,tf.stack
这篇文章主要介绍了浅谈tensorflow使用张量时的一些注意点tf.concat,tf.reshape,tf.stack,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06使用Python代码实现Linux中的ls遍历目录命令的实例代码
这次我就要试着用 Python 来实现一下 Linux 中的 ls 命令, 小小地证明下 Python 的不简单,需要的朋友可以参考下2019-09-09Python Social Auth构建灵活而强大的社交登录系统实例探究
这篇文章主要为大家介绍了Python Social Auth构建灵活而强大的社交登录系统实例探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2024-01-01tensorflow saver 保存和恢复指定 tensor的实例讲解
今天小编就为大家分享一篇tensorflow saver 保存和恢复指定 tensor的实例讲解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2018-07-07
最新评论