pytorch 多维矩阵索引实际上是如何工作的?

h7appiyu  于 2023-08-05  发布在  其他
关注(0)|答案(1)|浏览(85)

很难写这个标题,抱歉。
我看了一个视频,看到我可以用矩阵X索引矩阵C(下图)。但我不明白这里的机械师。由于我缺乏理解,我试图用更小的矩阵来重现它,只是为了让自己形象化......但我不能。
这里的CX矩阵与numsindexers矩阵有什么不同?在这两种情况下,大小是不同的,索引矩阵大于被索引的矩阵,并且索引矩阵中的值与索引矩阵中的值不同:


的数据

d4so4syb

d4so4syb1#

我想,我已经想明白了。索引矩阵的元素值必须在要索引的矩阵中要索引的维的长度之内。


的数据
indexers值将用于索引nums的行维度,其长度为2。这意味着索引(indexers矩阵中的值)必须为0或1。

相关问题