14_pytorch.where,pytorch.gather

我会带着你远行 2022-10-26 08:09 294阅读 0赞

## 1.14.高级OP ##

### 1.14.1.where ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdG8xMjk3NDg4NTA0_size_16_color_FFFFFF_t_70]  
给定一个条件**cond**，满足条件的**取x对应位置元素**，不满足的取**y对应元素**。

**代码示例：**

# -*- coding: UTF-8 -*-
    
    import torch
    
    a = 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 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdG8xMjk3NDg4NTA0_size_16_color_FFFFFF_t_70 1]**举例**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdG8xMjk3NDg4NTA0_size_16_color_FFFFFF_t_70 2]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdG8xMjk3NDg4NTA0_size_16_color_FFFFFF_t_70]: /images/20221024/f55be578b4a94c72b92df4f8b41e2d9e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdG8xMjk3NDg4NTA0_size_16_color_FFFFFF_t_70 1]: /images/20221024/6a338c351afa430e8a717d120439c93a.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdG8xMjk3NDg4NTA0_size_16_color_FFFFFF_t_70 2]: /images/20221024/12af36aa7a9e409fa3359a944cce8804.png

发表评论取消回复

表情：

评论列表（有 0 条评论，294人围观）

还没有评论，来说两句吧...

相关阅读

相关 java classes14下载_JAVA Class14

packagecom.store.view;importjava.util.ArrayList;importjava.util.Scanner;importcom.store.

落日映苍穹つ/ 2022年11月03日 01:38/ 0 赞/ 186 阅读

相关 14、MODULE

CLASS torch.nn.Module 所有神经网络模块的基类。你的模型也应该继承这个类。模块还可以包含其他模块，允许将它们嵌套在树结构中。您可以将子模块分配为

Love The Way You Lie/ 2022年08月31日 04:43/ 0 赞/ 222 阅读

相关 leetcode14

Longest Common Prefix 也是看题目类似，就跳着做题。题目就是在以string组成的数组中找这些string的最长前缀。我的思路： 1、第一步先对

女爷i/ 2022年08月09日 00:47/ 0 赞/ 186 阅读

相关 ex14

package exercise; import java.awt.BorderLayout; import java.awt.event.A

￡神魔★判官ぃ/ 2022年08月04日 13:37/ 0 赞/ 197 阅读

相关 java学习（14）

java学习（14） > 这篇接着来写一些Io流 > > > 1.LineNumberReader：之前写过的IO流的可以读写文件，而LineNumberRead

喜欢ヅ旅行/ 2022年06月15日 19:59/ 0 赞/ 195 阅读

相关 Intellij 14快捷键

转载自http://www.wolfbe.com/detail/201608/283.html 摘要：IDEA 全称IntelliJ IDEA，是java语言开发的集成环境，

忘是亡心i/ 2022年05月21日 00:30/ 0 赞/ 234 阅读

相关 14. 水晶球

Description 和许多同龄女孩子一样，久莲也喜欢水晶球。还有天，就是心心念念的他生日了。久莲希望把全世界最大最好看的水晶球送给他。她找到了宝石收藏家亚

浅浅的花香味﹌/ 2022年04月10日 04:23/ 0 赞/ 257 阅读

相关 14 AutoCompleteTextView

1.随笔提示：可以依靠 AutoCompleteTextView 实现；定义： java.lang

一时失言乱红尘/ 2022年03月19日 04:12/ 0 赞/ 218 阅读

相关 4.14

[www.cnblogs.com/cicaday/p/p…][www.cnblogs.com_cicaday_p_p] 装饰器 [blog.csdn.net/poison6/

柔情只为你懂/ 2022年01月20日 08:51/ 0 赞/ 267 阅读

相关 LeetCode 14

问题描述：编写一个函数来查找字符串数组中的最长公共前缀。如果不存在公共前缀，返回空字符串 ""。示例 1: 输入: \["flower","flow","flig

阳光穿透心脏的1/2处/ 2021年12月19日 14:59/ 0 赞/ 306 阅读