where() 条件判断 gather()查表 input ：待查的表 dim ： 在input的哪个维度上查表 index： 待查表的索引值 ...

1.Broadcasting Broadcasting能够实现Tensor自动维度增加（unsqueeze）与维度扩展（expand），以使两个Tensor的shape一致，从而完成某些操作，主要按照如下步骤进行： 从最后面的维度开始匹配（一般后面理解为小维度）； 在前面插入若干维度 ...

一、Pytorch安装 安装cuda和cudnn，例如cuda10，cudnn7.5 官网下载torch：https://pytorch.org/ 选择下载相应版本的torch 和torchvision的whl文件 使用pip install whl_dir安装torch，并且同时安装 ...

先放一张表，可以看成是二维数组 行(列)索引 索引0 索引1 索引2 索引3 索引0 0 ...

PyTorch中scatter和gather的用法 闲扯 许久没有更新博客了，2019年总体上看是荒废的，没有做出什么东西，明年春天就要开始准备实习了，虽然不找算法岗的工作，但是还是准备在2019年的最后一个半月认真整理一下自己学习的机器学习和深度学习的知识。 scatter的用法 ...

首先，给出官方文档的链接： https://pytorch.org/docs/stable/generated/torch.gather.html?highlight=gather#torch.gather 然后，我用白话翻译一下官方文档。 gather，顾名思义，聚集 ...

看得不明不白（我在下一篇中写了如何理解gather的用法） gather是一个比较复杂的操作，对一个2维tensor，输出的每个元素如下： 二维tensor的gather操作 针对0轴 注意index此时的值 输入 输出 分割线============ 针对1轴 ...

gather函数的的官方文档： 例子： 输出如下： 在上面的例子中，a是一个4×4矩阵： 1）当维度dim=0，索引index_1为[3,2,1,0]时，此时可将a看成1×4的矩阵，通过index_1对a每列进行行索引：第一列第四行元素为12，第二列第三行元素为9，第三列第二行 ...