14_pytorch.where,pytorch.gather

1.14.高级OP

1.14.1.where

给定一个条件cond，满足条件的取x对应位置元素，不满足的取y对应元素。

代码示例：

# -*- coding: UTF-8 -*-import torcha = torch.randn(3, 5)
b = torch.ones(3, 5)print(a)
"""
输出结果：
tensor([[ 1.4446,  0.4373,  0.6730,  0.1549, -1.0664],[ 0.4106,  0.3604,  1.3719, -0.9672,  0.7382],[ 0.2602, -0.3008,  0.8865, -0.2711,  0.1028]])
"""
print(b)
"""
输出结果：
tensor([[1., 1., 1., 1., 1.],[1., 1., 1., 1., 1.],[1., 1., 1., 1., 1.]])
"""print(torch.where(a > 0, a, b))
"""
输出结果：
tensor([[1.4446, 0.4373, 0.6730, 0.1549, 1.0000],[0.4106, 0.3604, 1.3719, 1.0000, 0.7382],[0.2602, 1.0000, 0.8865, 1.0000, 0.1028]])
"""

1.14.2.gather

举例

14_pytorch.where,pytorch.gather相关推荐

实现大规模图计算的算法思路
分享嘉宾:徐潇然 Hulu 研究员编辑整理:莫高鼎出品平台:DataFunTalk 导读:2017年我以深度学习研究员的身份加入Hulu,研究领域包括了图神经网络及NLP中的知识图谱推理,其中我们 ...
gather torch_浅谈Pytorch中的torch.gather函数的含义
pytorch中的gather函数 pytorch比tensorflow更加编程友好,所以准备用pytorch试着做最近要做的一些实验. 立个flag开始学习pytorch,新开一个分类整理学习pyt ...
pytorch新手需要注意的隐晦操作Tensor,max,gather
文章移到:https://oldpan.me/archives/pytorch-rookie-care-tensor-max-gather pytorch中有很多操作比较隐晦,需要仔细研究结合一些例子 ...
Pytorch的使用：torch.gather函数
Pytorch的使用:torch.gather函数 **torch.gather()** 作用:方便从批量tensor中获取特定化维度指定索引下的数据,该索引往往是乱序的. 首先看一下官方文档中的3维 ...
pytorch中gather用法
pytorch中gather的用法 2维度tensor进行映射: 3维度tensor进行映射: gather其实是对input进行一种映射,index必须是 LongTensor格式. 2维度tens ...
Pytorch中gather函数的个人理解方法
之前一直理解不了Pytorch中gather的用法,看了官方的文档也是一头雾水.然后自己琢磨,找规律,用以下方法进行理解. 一.官方文档 torch.gather(input, dim, index, ...
Pytorch中的torch.gather函数的含义
pytorch中的gather函数 pytorch比tensorflow更加编程友好,所以准备用pytorch试着做最近要做的一些实验. 立个flag开始学习pytorch,新开一个分类整理学习pyt ...
Pytorch中scatter与gather操作
文章目录数据发散scatter 带聚集的发散scatter_add_ onnx中scatterND 数据聚集gather 数据发散scatter 函数原型pytorch官方文档scatter_: s ...
理解 PyTorch 中的 gather 函数
好久没更新博客了,最近一直在忙,既有生活上的也有工作上的.道阻且长啊. 今天来水一文,说一说最近工作上遇到的一个函数:torch.gather() . 文字理解我遇到的代码是 NLP 相关的,代码中 ...

14_pytorch.where,pytorch.gather

1.14.高级OP

1.14.1.where

1.14.2.gather

14_pytorch.where,pytorch.gather相关推荐

最新文章

热门文章