pytorch函数
文章平均质量分 54
qq_43675321
这个作者很懒,什么都没留下…
展开
-
[pytorch函数合集]torch.index_select()
如上如所示,第一个参数arr1是我们需要操作的数组;第二个参数是需要操作的维度,维度是0为按行操作,维度是1为按列操作;这次讲解的函数是scatter_max(),该函数有4个参数,其中前三个分别为,(1)将要被操作的目标数组arr1(2)设定操作数组arr1哪个维度的参数dim(3)目录函数(我自己起的名)index。原创 2023-09-13 20:38:18 · 108 阅读 · 1 评论 -
[pytorch函数合集]torch_scatter.scatter_max()
这次讲解的函数是scatter_max(),该函数有4个参数,其中前三个分别为,(1)将要被操作的目标数组arr1(2)将arr1数组进行分组的数组arr2(3)设定操作数组arr1哪个维度的参数dim(4)显示几组数据的参数dim_size。原创 2023-09-13 20:06:23 · 795 阅读 · 2 评论