From 6dff47ae7190ff57aeeea99d1d87269d500a0aad Mon Sep 17 00:00:00 2001 From: Tingyu Wang Date: Wed, 24 Jan 2024 11:19:45 -0500 Subject: [PATCH] Update python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py Co-authored-by: Mario Geiger --- .../cugraph_equivariant/nn/tensor_product_conv.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py b/python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py index 998b7013446..ea9979427d0 100644 --- a/python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py +++ b/python/cugraph-equivariant/cugraph_equivariant/nn/tensor_product_conv.py @@ -186,6 +186,7 @@ def forward( Reduction operator. Choose between "mean" and "sum". edge_envelope: torch.Tensor, optional + Optional, typically used as attenuation factors to fade out messages coming from nodes close to the cutoff distance used to create the graph. This is important to make the model smooth to the changes in node's coordinates. Edge envelope. Shape: (num_edges,)