原文:Pytorch中的torch.gather函数

gather函数的的官方文档: 例子: 输出如下: 在上面的例子中,a是一个 矩阵: 当维度dim ,索引index 为 , , , 时,此时可将a看成 的矩阵,通过index 对a每列进行行索引:第一列第四行元素为 ,第二列第三行元素为 ,第三列第二行元素为 ,第四列第一行元素为 ,即b , , , 当维度dim ,索引index 为 , , , T时,此时可将a看成 的矩阵,通过index 对 ...

2020-08-30 15:48 0 633 推荐指数:

查看详情

pytorch gather() 函数详解

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

Mon Feb 15 22:43:00 CST 2021 0 700
Pytorchtorch.where函数

首先我们看一下Pytorchtorch.where函数是怎样定义的: torch.where函数的功能如下: 以具体实例看一下torch.where函数的效果: 结果如下: 可以看到torch.where函数会对condition ...

Wed Dec 29 23:17:00 CST 2021 0 1044
pytorchtorch.narrow()函数

torch.narrow(input, dim, start, length) → Tensor Returns a new tensor that is a narrowed version of input tensor. The dimension dim is input from ...

Fri Nov 15 05:18:00 CST 2019 0 1238
Pytorchtorch.cat()函数

以把list的tensor拼接起来。 比如: 上面的代码可以合成一行来写: ...

Mon Jul 15 23:03:00 CST 2019 0 1921
Pytorchtorch.cat()函数

cat是concatnate的意思:拼接,联系在一起。 先说cat( )的普通用法 如果我们有两个tensor是A和B,想把他们拼接在一起,需要如下操作: 其次,cat还可以把list的tensor拼接起来。 比如: 上面的代码可以合成 ...

Sun Dec 23 04:29:00 CST 2018 0 43400
PyTorchscatter和gather的用法

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

Sun Nov 17 19:21:00 CST 2019 0 457
pytorchtorch.cat(),torch.chunk(),torch.split()函数的使用方法

一、torch.cat()函数 熟悉C字符串的同学们应该都用过strcat()函数,这个函数在C/C++程序中用于连接2个C字符串。在pytorch,同样有这样的函数,那就是torch.cat()函数. 先上源码定义:torch.cat(tensors,dim=0,out=None ...

Mon Apr 13 00:09:00 CST 2020 2 6877
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM