np.where和torch.where的使用

最新推荐文章于 2024-06-02 10:39:13 发布
最新推荐文章于 2024-06-02 10:39:13 发布
阅读量1.9k 收藏
点赞数
分类专栏： python问题 文章标签： python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_40426830/article/details/108711765
版权

np.where和torch.where的使用

两者都是三个输入参数，第一个是判断条件，第二个是符合条件的设置值，第三个是不满足条件的设置值。

区别在于，torch要将设置值全部改为Tensor类型的

#预测概率
y_pred_probs = net(torch.tensor(x_test[0:10]).float()).data
y_pred_probs

在这里插入图片描述

#预测类别
y_pred = torch.where(y_pred_probs>0.5,
torch.ones_like(y_pred_probs),torch.zeros_like(y_pred_probs))
y_pred

在这里插入图片描述

y_pred = np.where(np.isnan(y_pred_probs), 0, 1)
y_pred

在这里插入图片描述

风雪云侠
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
基于torch.where和布尔索引的速度比较
09-18
今天小编就为大家分享一篇基于torch.where和布尔索引的速度比较，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
yolov5s.torchscript.pt
10-13
yolov5 v3.0版本，根据yolov5s.pt权重处理的，适合在移动端推理的torchscript序列化文件。
torch.where 与 np.where
蓝羽飞鸟的博客
03-24 618
numpy.where如果不指定条件，返回的是>0元素的位置（下标），见链接 但是如果torch.where不指定条件，返回的是非0元素的位置（下标），包括了负数。 看个例子 >>> c tensor([[[ 0., 0.], [-1., -1.], [-1., -2.]], [[ 1., 1.], [ 0., 0.], [ 0., -1.]], [[ 1., 2.],
np.where 和 torch.where 用法
weixin_43102043的博客
06-30 506
numpy.where() 有两种用法： numy.where(condition) 只有condiiton，输出满足condition的元素的下标, 这里的坐标以tuple的形式给出，通常原数组有多少维，输出的tuple中就包含几个数组，分别对应符合条件元素的各维坐标 torch.where torch.where()函数的作用是按照一定的规则合并两个tensor类型...
PyTorch学习（10）：torch.where
最新发布
tecsai的博客
06-02 464
在PyTorch中，torch.where 函数是一种强大的工具，用于根据条件从两个张量中选择元素。这个函数在数据处理和神经网络的正则化技术中特别有用，比如实现dropout或者masking策略。torch.where 的基本用法类似于Python的内建函数 np.where，但它专门针对PyTorch张量进行了优化。
np.where和torch.where的使用区别
Nick Blog
01-22 3万+
两者都是三个输入参数，第一个是判断条件，第二个是符合条件的设置值，第三个是不满足条件的设置值。 区别在于，torch要将设置值全部改为Tensor类型的 mask_true = torch.where(torch.isnan(y_true), torch.full_like(y_true, 0), torch.full_like(y_true, 1)) mask_true = np.w...
torch.where(),np.where()的两种用法，以及np.argwhere()寻找张量（tensor）和数组中为0的索引
weixin_45791990的博客
03-13 3479
torch.where(),np.where()的两种用法，以及np.argwhere()寻找张量(tensor)和数组中为0的索引
torch.where()函数
StriveHeisenberg的博客
11-07 2264
Pytorch的torch.where（）函数和numpy中的np.where（）函数使用方法完全是一致的。torch.where（）需要三个参数：torch.where（第一个为判断条件，第二个为满足条件设置的值，第三个为不满足条件设置的值）。等同于一个基本判断结构： if 条件： 值1 else： 值2 example： y_pred = torch.where(y_pred > 0.5, torch.ones_like(y_pred, dtype=torch.float32),
PyTorch的torch.cat用法
09-16
下面我们将详细探讨`torch.cat`的使用方法、注意事项以及其在深度学习中的应用。 1. **基本概念**： - `torch.cat`是`concatenate`的缩写，它的主要功能是将两个或多个张量沿着指定的维度合并成一个新的张量。 - ...
python torch.utils.data.DataLoader使用方法
09-17
主要介绍了python torch.utils.data.DataLoader使用方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
torch.where和numpy.where faster element wise operation build-in implement
WYXHAHAHA123的博客
01-24 488
import torch a=torch.randn((2,3)) print(a) ''' tensor([[-0.4695, 0.0949, -0.7063], [-0.2190, 0.7319, -0.6530]]) ''' b=torch.where(a>0,torch.tensor(10.),a) print(b) ''' tensor([[-0.4695, ...
torch.where() 与 numpy.where() 用法详解
知识搬运者
01-12 1979
1、torch.where() torch.where(condition，a，b)。其中，输入参数condition：条件限制，如果满足条件，则选择a，否则选择b作为输出。a和b是 tensor。 import torch a = torch.tensor([[0.0349, 0.0670, -0.0612, 0.0280, -0.0222, 0.0422], [-1.6719, 0.1242, -0.6488, 0.3313, -1.3965, -0.068...
np.where 的用法记录
Lily的博客
10-29 364
np.where 的用法
torch.where和布尔索引的速度比较
WYXHAHAHA123的博客
03-05 3014
import torch import time x = torch.Tensor([[1, 2, 3], [5, 5, 5], [7, 8, 9],[5,5,5],[1,2,3,],[1,2,4]]) ''' 使用pytorch实现对于任意shape的torch.tensor,如果其中的element不等于5则为0，等于5则保留原数值 实现该功能的两种方式，并比较两种实现方式的速度 ''' ...
torch.where的使用
spark的博客
05-12 453
如果b_bool是tensor类型，那返回的元组含有的唯一元素就是所有true角标的tensor。在这个语句中a会提取出b_bool为true的角标，但会返回成一个元组。函数返回一个张量或元组，具体输出是由函数输出参数的数量和类型所决定的。所有为了a直接得到一个tensor，我们用【0】做索引。例，在下面的示例中，我们用另一个张量。在 PyTorch 中，
torch.where()的用法以及例子
凝眸伏笔的博客
01-09 5万+
1.用法 torch.where()函数的作用是按照一定的规则合并两个tensor类型。 torch.where(condition，a，b)其中 输入参数condition：条件限制，如果满足条件，则选择a，否则选择b作为输出。 注意：a和b是tensor. 2.例子 import torch a = torch.tensor([[0.0349, 0.0670, -0.0612, 0.0280, -0.0222, 0.0422], [-1.6719, 0.1242,
torch.where()函数解读
qq_40178291的博客
10-12 5万+
函数作用 torch.where()函数的作用是按照一定的规则合并两个tensor类型。 代码示例 >>> import torch >>> a=torch.randn(3,5) >>> b=torch.ones(3,5) >>> a tensor([[-0.0310, 1.5895, 1.6003, -1.7584,...
pytorch torch.where用法
Orientliu96的博客
03-12 3万+
pytorch torch.where()函数用法 torch.where(condition, x, y) → Tensor 函数的作用 根据条件，返回从x,y中选择元素所组成的张量。如果满足条件，则返回x中元素。若不满足，返回y中元素。 例子 ...
np.where()用法总结
热门推荐
棒子皮蹦蹦床
03-21 10万+
用法一：可以返回一个n维数组，可广播。# np.where(condition, x, y)# condition, x, y, return are all can be #ndarray x = np.random.randn((4, 4)) np.where(x>0, 2, -2) output:用法二：condition 也可以是布尔型数组，每个条件都和x,y对应（广播）用法三：找...
torch.where np.where
09-18
在NumPy中，np.where()函数的功能与torch.where()类似，但输入和输出的类型不同。np.where()函数接受一个条件和一个数组作为输入，并返回与条件为True的元素对应的索引。例如，如果我们有一个数组arr，我们可以使用...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值