(A, X, r)
| 19 | # diffusion in SIGN laconically. |
| 20 | ################################################################################ |
| 21 | def sign_diffusion(A, X, r): |
| 22 | # Perform the r-hop diffusion operation. |
| 23 | X_sign = [X] |
| 24 | for _ in range(r): |
| 25 | X = A @ X |
| 26 | X_sign.append(X) |
| 27 | return X_sign |
| 28 | |
| 29 | |
| 30 | class SIGN(nn.Module): |