Graph operationsΒΆ

radius_graph(positions, radius[, batch])

Build the directed, loop-free batched radius graph used by gate-points models.

scatter_sum(source, index[, dim_size, ...])

Sum rows of source into rows selected by one-dimensional index.