最近在学习pytorch函数时需要做独热码,然后遇到了scatter_()函数,不太明白意思,现在懂了记录一下以免以后忘记。 这个函数是用一个src的源张量或者标量以及索引来修改另一个张量。这个函数主要有三个参数scatter_(dim,index,src) dim:沿着哪个维度来进行索引(一会儿举个例子就明白了) index:用来进行索引的张量 src:...
以下内容,转自daremosiranaihana这位博主的博客,地址为https://www.cnblogs.com/daremosiranaihana/p/12538512.html 最近在学习pytorch函数时需要做独热码,然后遇到了scatter_()函数。 这个函数是用一个src的源张量或者标量以及索引来修改另一个张量。这个函数主要有三个参数scatter_(dim,index,src) dim:沿着哪个维度...
PyTorch 中,一般函数加下划线代表直接在原来的 Tensor 上修改 scatter(dim, index, src) 的参数有 3 个 dim:沿着哪个维度进行索引 index:用来 scatter 的元素索引 src:用来 scatter 的源元素,可以是一个标量或一个张量 这个scatter可以理解成放置元素或者修改元素 简单说就是通过一个张量 src 来修改另一个张量,...
pytorch中的scatter_()函数 2020-03-21 12:18 −... daremosiranaihana 0 3322 pytorch调用caffe的lmdb 2019-11-21 14:20 −代码:https://github.com/liangX-box/pytorchReadLmdb.git 一. 处理好训练集和验证集后,通过caffe的convert_imageset生成lmdb: 1 /usr/softwares/caffe/build/tools/convert_images...
在PyTorch 的scatter函数中,当你指定dim=0时,index张量确实是指定了src中每个元素应该放置在res张量的哪一行。不过,列号的选择是由index张量的形状和src张量的形状共同决定的。 1. 基本逻辑 dim=0表示scatter操作将在行方向上进行。 index张量的形状应该与src张量的形状匹配,或者能够广播到相同的形状。