torch.where()的用法以及例子

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, -0.6488, 0.3313, -1.3965, -0.0682],[-1.3419,  0.4485, -0.6589, 0.1420, -0.3260, -0.4795]])
b = torch.tensor([[-0.0658, -0.1490, -0.1684, 0.7188,  0.3129, -0.1116],[-0.2098, -0.2980,  0.1126, 0.9666, -0.0178,  0.1222],[ 0.1179, -0.4622, -0.2112, 1.1151,  0.1846,  0.4283]])
cc = torch.where(a>0,a,b)     #合并a,b两个tensor，如果a中元素大于0，则c中与a对应的位置取a的值，否则取b的值
print(cc)

3.numpy的where()，这里的输入非tensor。

a = np.arange(10)
print(a)
c = np.where(a%2==0, a, 10 * a)
print(c)

torch.where()的用法以及例子相关推荐

pytorch torch.sort()的用法以及例子
1.作用对tensor中元素排序 2.用法 dim = -1,按照行排序,dim= 1按照列排序,descending=True,则递减排序,否则递增 3.例子按照行排序 logits = tor ...
conv2d的输入_pytorch1.0中torch.nn.Conv2d用法详解
Conv2d的简单使用 torch 包 nn 中 Conv2d 的用法与 tensorflow 中类似,但不完全一样. 在 torch 中,Conv2d 有几个基本的参数,分别是 in_channel ...
torch.rand、torch.randn及torch.normal的用法
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 torch.rand.torch.randn及torch.normal的用法一.torch.rand 二.torch.randn 三 ...
QSettings 用法（例子）
QSettings 用法(例子) 需求分析日常使用中,程序重新启动的时候,希望记住软件上次使用关闭之前的一些状态. QSettings能记住软件的信息,并保存在操作系统的注册表中,比如windows ...
torch.nn.Conv2d() 用法讲解
本文是深度学习框架 pytorch 的API : torch.nn.Conv2d() 函数的用法.本博客介绍了 torch.nn.Conv2d() 各个参数的含义和用法,学会使用 pytorch 创建 ...
torch.sum()的用法
torch.sum()对输入的tensor数据的某一维度求和,一共两种用法 1．torch.sum(input, dtype=None) 2．torch.sum(input, list: dim, b ...
pytorch学习中 torch.squeeze() 和torch.unsqueeze()的用法
squeeze的用法主要就是对数据的维度进行压缩或者解压. 先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的 ...
pytorch中的torch.tensor.repeat以及torch.tensor.expand用法
文章目录 torch.tensor.expand torch.tensor.repeat torch.tensor.expand 先看招 import torch x = torch.tensor([ ...
linux GDB 简单用法以及例子
首先GDB是类unix系统下一个优秀的调试工具, 当然作为debug代码的工具, 肯定没有IDE来的直观了. 不过, 命令行自然有命令行的有点, 当你无法是使用IDE时, gdb有时会帮上 ...

torch.where()的用法以及例子

1.用法

2.例子

3.numpy的where()，这里的输入非tensor。

torch.where()的用法以及例子相关推荐

最新文章

热门文章