最近在学习pytorch函数时需要做独热码,然后遇到了scatter_()函数,不太明白意思,现在懂了记录一下以免以后忘记。 这个函数是用一个src的源张量或者标量以及索引来修改另一个张量。这个函数主要有三个参数scatter_(dim,index,src) dim:沿着哪个维度来进行索引 ...
https: blog.csdn.net qq article details scatter input, dim, index, src 将src中数据根据index中的索引按照dim的方向填进input中. dim ,分别对每列填充: 实现原理: 对于LoneTensor内的矩阵,暂且称为tmp , , , , , , , , , 将最终的 的矩阵,暂且称为result。result初始为全 ...
2019-05-25 22:35 0 4839 推荐指数:
最近在学习pytorch函数时需要做独热码,然后遇到了scatter_()函数,不太明白意思,现在懂了记录一下以免以后忘记。 这个函数是用一个src的源张量或者标量以及索引来修改另一个张量。这个函数主要有三个参数scatter_(dim,index,src) dim:沿着哪个维度来进行索引 ...
scatter() 和 scatter_() 的作用是一样的,只不过 scatter() 不会直接修改原来的 Tensor,而 scatter_() 会 PyTorch 中,一般函数加下划线代表直接在原来的 Tensor 上修改 scatter(dim, index, src ...
看得不明不白(我在下一篇中写了如何理解gather的用法) gather是一个比较复杂的操作,对一个2维tensor,输出的每个元素如下: 二维tensor的gather操作 针对0轴 ...
PyTorch中scatter和gather的用法 闲扯 许久没有更新博客了,2019年总体上看是荒废的,没有做出什么东西,明年春天就要开始准备实习了,虽然不找算法岗的工作,但是还是准备在2019年的最后一个半月认真整理一下自己学习的机器学习和深度学习的知识。 scatter的用法 ...
绘制数据图 散布图(scatter plots) 需要掌握的命令: plot() ...
pytorch笔记 <三> optimizer.zero_grad() 将梯度变为0,用于每个batch最开始,因为梯度在不同batch之间不是累加的,所以必须在每个batch开始的时候初始化累计梯度,重置为0. torch.max() 在某个dim上返回最大的值 ...
pytorch笔记 - torchvision.utils.make_grid torchvision.utils.make_grid 怎么理解这个输出结果呢?第一个dim当然就是channel,因为合并成一张图片了嘛,所以batch这个维度就融合了,变成了chw,这里c还是原来 ...
原文地址:https://pytorch.org/tutorials/beginner/deep_learning_60min_blitz.html 什么是pytorch? pytorch是一个基于python语言的的科学计算包,主要分为两种受众: 能够使用GPU运算取代 ...