Initial commit

b73cc994 · quyuanhao123 · b73cc994
Commit b73cc994 authored Apr 18, 2023 by quyuanhao123
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

torch_spline_conv/weighting.py torch_spline_conv/weighting.py +9 -0

No files found.
--- a/torch_spline_conv/weighting.py
+++ b/torch_spline_conv/weighting.py
+import torch
+
+
+@torch.jit.script
+def spline_weighting(x: torch.Tensor, weight: torch.Tensor,
+                     basis: torch.Tensor,
+                     weight_index: torch.Tensor) -> torch.Tensor:
+    return torch.ops.torch_spline_conv.spline_weighting(
+        x, weight, basis, weight_index)