当前位置:网站首页>tensor.eq() tensor.item() tensor.argmax()
tensor.eq() tensor.item() tensor.argmax()
2022-08-09 10:28:00 【白十月】
tensor.eq()
torch.eq(input, other, *, out=None)
对两个张量Tensor进行逐元素的比较,若相同位置的两个元素相同,则返回True;若不同,返回False。
Parameters(参数):
input :必须是一个Tensor,该张量用于比较
other :可以是一个张量Tensor,也可以是一个值value
return(返回值):返回一个Boolean类型的张量,对两个张量Tensor进行逐元素的比较,若相同位置的两个元素相同,则返回True;若不同,返回False。
import torch
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[1, 1], [3, 3]])
print(x)
print(y)
out = torch.eq(x, y)
print(out)
out1 = torch.eq(x, y).sum()
print(out1)
out2 = torch.eq(x, y).sum().float()
print(out2)
out3 = torch.eq(x, y).sum().item()
print(out3)
输出为:
tensor([[1, 2],
[3, 4]])
tensor([[1, 1],
[3, 3]])
tensor([[ True, False],
[ True, False]])
tensor(2)
tensor(2.)
2
tensor.item()
item() 返回一个数
该方法的功能是以标准的Python数字的形式来返回这个张量的值.这个方法
只能用于只包含一个元素的张量.对于其他的张量,请查看方法tolist().
tensor.argmax()
返回输入张量中所有元素的最大值的索引。
官方文档:https://pytorch.org/docs/stable/generated/torch.argmax.html
import torch
a = torch.randn(3, 4)
torch.argmax(a)
torch.argmax(a,dim=1)
torch.argmax(a,dim=0)
tensor([[-0.0815, 0.8759, -0.1932, 0.6896],
[-1.1432, -0.4114, -0.6331, 0.2309],
[-0.2122, 0.2187, 0.3195, -0.7594]])
tensor(1)
tensor([1, 3, 2])
tensor([0, 0, 2, 0])
边栏推荐
猜你喜欢
【MySQL】mysql因为字符集导致left join出现Using join buffer (Block Nested Loop)
深度学习--生成对抗网络(Generative Adversarial Nets)
[Halcon&定位] 解决Roi区域外的模板匹配成功
史上最小白之《Word2vec》详解
一天半的结果——xmms on E2
机器学习-逻辑回归(logistics regression)
深度学习--循环神经网络(Recurrent Neural Network)
shell脚本实战(第2版)/人民邮电出版社 脚本2 验证输入:仅限字母和数字
The common problems in laptops, continuously updated
拿下跨界C1轮投资,本土Tier 1高阶智能驾驶系统迅速“出圈”
随机推荐
借问变量何处存,牧童笑称用指针,Go lang1.18入门精炼教程,由白丁入鸿儒,go lang类型指针(Pointer)的使用EP05
hover内部指定子类的样式
抛出一个问题? Mysql环境下进行Count操作执行的时候速度很慢_需手动给主键添加索引---MySql优化001
壁纸
判断一段文字的width
MySQL全文索引
实现下拉加载更多
深度学习--生成对抗网络(Generative Adversarial Nets)
程序员的专属浪漫——用3D Engine 5分钟实现烟花绽放效果
unix环境编程 第十四章 14.4 I/O多路转接
单元测试2之实际结果检查的引用
antd的Table列选择、列拓展
MySQL和MyEclipse的数据库连接操作
关于页面初始化
UNIX Environment Programming Chapter 15 15.5FIFO
1001 害死人不偿命的(3n+1)猜想 (15 分)
第二周作业
Attentional Feature Fusion
OpenGL 2.0编程例子
xmms的均衡器试验成功 音效相当不错 比rockbox可能还要好