torch.where(condition,x,y)

于 2023-12-06 15:16:46 发布
阅读量190 收藏
点赞数 5
文章标签： pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_39710897/article/details/134832415
版权

会根据condition的不同返回x或y，当满足条件是返回x，否则返回y。

p.s. x和y都是tensor形式。

========分割线========

以下内容摘自Pytorch官方文档：

NOTE

（1）The tensors condition, x, y must be broadcastable.

（2）Currently valid scalar and tensor combination are 1. Scalar of floating dtype and torch.double 2. Scalar of integral dtype and torch.long 3. Scalar of complex dtype and torch.complex128

Parameters

  • condition (BoolTensor) – When True (nonzero), yield x, otherwise yield y

  • x (Tensor or Scalar) – value (if x is a scalar) or values selected at indices where condition is True

  • y (Tensor or Scalar) – value (if y is a scalar) or values selected at indices where condition is False

Returns：A tensor of shape equal to the broadcasted shape of condition, x, y

不会敲代码综合征
关注
  • 5
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
【Torch API】pytorch中的torch.where()函数详解
devil_son1234的博客
01-04 1万+
pytorch中的torch.where()函数详解
torch.where函数用法
jmhIcoding
04-13 2091
函数原型：torch.where(condition, x, y) → Tensor 作用： 将两个tensor ： x和y进行逐元素合并，condition ，x和y需要是相同形状的、或者可以广播为相同形状。 假设合并后的结果为z，z的形状为x，y两者的形状或者他们广播后的形状。 对于每个下标i，如果condition[i] 满足条件，那么z[i]等于x[i]，否则z[i]等于y[i] 。 举例： >>> x = torch.randn(3, 2) >>> y = to
【材料整理】 -- torch中常用知识点。
银尘博客
03-27 1470
文章目录 文章目录写在前面import torch 写在前面 本文主要总结自己毕设过程中，用到torch库中一些知识点，之后会不断补充，欢迎大家收藏！ import torch 1、torch.where函数 跟双目运算符一样； torch.where(condition,a,b) condition为真，执行a；否则，执行b。 2、torch.numpy()将Tensor变量转换为ndarray变量，其中t是一个Tensor变量，可以是标量，也可以是向量，转换后dtype与Tensor的dtype一
torch.where()的用法
m0_53501972的博客
12-03 1670
pytorch初学者
Condition的使用和理解
qq_36794621的博客
03-12 466
接上一篇AQS,所以编号从四开始 四，Condition 在前面学习 synchronized 的时候，有讲到 wait/notify 的基本使用，结合 synchronized 可以实现对线程的通信。那么这个时候我就在思考了，既然 J.U.C 里 面提供了锁的实现机制，那 J.U.C 里面有没有提供类似的线程通信的工具呢？ 于 是找阿找，发现了一个 Condition 工具类。 Conditio...
x = torch.where(torch.isnan(x), torch.zeros_like(x), x)
05-05
最后，torch.where(condition, x, y) 函数会根据 condition 张量的值，在 x 和 y 之间进行选择：当 condition 的值为 True 时，选择 x 中对应位置的值；当 condition 的值为 False 时，选择 y 中对应位置的值。因此...
Pytorch学习笔记（一）torch.where()：根据条件修改张量
weixin_42860775的博客
05-06 1059
条件判断时，张量特定值的修改。运用到了torch.where()的方法。
torch.where(x, x, 2x)
05-14
这是一个使用 PyTorch 中的 torch.where 函数的示例，其中 x 是一个张量。 torch.where 函数的作用是根据条件选择张量中的元素。它接受三个参数：condition，x 和 y。其中 condition 是一个布尔型的张量，与 x 和 y...
torch.where
04-30
torch.where(condition, x, y) ``` 其中，`condition`是一个与`x`和`y`具有相同形状的布尔张量，表示要选择哪些元素。如果`condition`中的元素为`True`，则选择`x`中对应的元素，否则选择`y`中对应的元素。 下面是...
c++学习笔记
最新发布
m0_54720387的博客
09-25 1159
c++初学者学习c++
详解PyTorch中Tensor的高阶操作
01-20
条件选取：torch.where(condition, x, y) → Tensor 返回从 x 或 y 中选择元素的张量，取决于 condition 操作定义： 举个例子： >>> import torch >>> c = randn(2, 3) >>> c tensor([[ 0.0309, -1.5993, 0.1986], [-0.0699, -2.7813, -1.1828]]) >>> a = torch.ones(2, 3) >>> a tensor([[1., 1., 1.], [1., 1., 1.]]) >>> b = torch.zeros(2,
[学习笔记-C++篇]day1 输入输出-判断
warningm_dm的博客
03-06 532
C++自学（真）指南（不适用别人）
高质量C++编程指南 一
nowgoant的专栏
05-06 449
<br /><br />第1章 文件结构<br />1.4 头文件的作 用<br />早期的编程语言如Basic、Fortran没有头文件的概念，C++/C语言的初学者虽然会用使用头文件，但常常不明其理。这里对头文件的作用略作解释：<br />（1）通过头文件来调用库功能。在很多场合，源 代码不便（或不准）向用户公布，只要向用户提供头文件和二进制的库即可。用户只需要按照头文件中的接口声明来调用库功能，而不必关心接口怎么实现的。编译 器会从库中提取相应的代码。<br />（2）头文件能加强类型安全检查。如果某
torch.where用法
热门推荐
jacke121的专栏
09-22 2万+
  把矩阵x中大于5的变成5了 import torch x = torch.linspace(1, 27, steps=27).view(9, 3) bbb = torch.where(x &gt; 5, torch.full_like(x, 5), x) print(bbb)
torch 常用类与方法
yichudu
11-23 3104
dataset 通过 from torch.utils.data import IterableDataset, DataLoader 引入. 典型用法如 loader=DataLoader(dataset:Dataset), 二者一起等价于 tf 的 dataset. 流式数据 典型场景是内存盛不下, 网络数据库->dataset->model feed 流式运作. 例子见下. from torch.utils.data import IterableDataset, DataLoader
Pytorch 一些高阶操作
洪流之源
06-26 1191
1.where 使用C=torch.where(condition,A,B)其中A,B,C,condition是shape相同的Tensor，C中的某些元素来自A，某些元素来自B，这由condition中对应位置的元素是1还是0来决定。如果condition对应位置元素是1，则C中的该位置的元素来自A中的该位置的元素，如果condition对应位置元素是0，则C中的该位置的元素来自B中的该位置...
torch.where()的两种用法
zy_join的博客
10-19 2987
torch.where函数的用法;1.合并两个tensor 2.取某像素值的坐标
torch.where和布尔索引的速度比较
WYXHAHAHA123的博客
03-05 3045
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()函数详解
lxhRichard的博客
11-20 2625
最近查了一些关于 torch.where() 函数的相关解释，有的高赞说torch.where()函数的作用是按照一定的规则合并两个tensor类型。其实合并这个词用的很不恰当，解释的很牵强。通过查找官方文档，整理一下对这个函数进行解释。
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值