pytorch每日一学49(torch.where())根据t指定条件更改指定tensor中的数值

这篇博客详细介绍了PyTorch中的torch.where函数,它根据条件选择张量中的元素。当条件满足时,保留原张量的值,否则替换为另一个张量的对应值。文章通过例子展示了如何使用该函数,并指出张量与标量的组合要求。此外,还提到了torch.nonzero(condition, astuple=True)与torch.where(condition)在查找满足条件元素位置上的等价性。

第49个方法

torch.where(condition, x, y) → Tensor

此方法是将x中的元素和条件相比,如果符合条件就还等于原来的元素,如果不满足条件的话,那就去y中对应的值,公式为
在这里插入图片描述
举个例子就清楚了,官方例子如下:

>>> x = torch.randn(3, 2)
>>> y = torch.ones(3, 2)
>>> x
tensor(
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值