How torch.ops.pyg.segment_matmul works? #6322

Unanswered

1049451037 asked this question in Q&A

1049451037
Dec 31, 2022

Hi, there is a line of code in pyg_lib:

return torch.ops.pyg.segment_matmul(inputs, ptr, other)

How does this work? Where can I find the source code of this function?

Replies: 1 comment

LukeLIN-web
Jan 1, 2023

See https://github.com/pyg-team/pyg-lib/blob/master/pyg_lib/csrc/ops/matmul.h

0 replies

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment