第五节作业 理解采样函数时的困惑

我观察到是Graph的sample_adj方法实现了采样 当size不为-1时通过indptr, indices = self._sample_adj(len(batch), indices, indptr, size)得到所有采样点的csr索引
但是我不太理解这段代码在做什么 并且我认为传入的indices是一维的 rand是二维的 所以维度相关的操作我有些困惑 一定是我哪里理解错了 :melting_face:

如果传入的indices是一维的,而rand是二维的,在进行索引操作时会自动拓展维度和rand的维度对齐

1 个赞