Skip to content

Commit b6f7c17

Browse files
AleDinvedario-coscia
authored andcommitted
fix radial field
1 parent 9269702 commit b6f7c17

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pina/model/block/message_passing/radial_field_network_block.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,8 @@ def forward(self, x, edge_index):
105105
106106
:param x: The node features.
107107
:type x: torch.Tensor | LabelTensor
108-
:param torch.Tensor edge_index: The edge indices.
108+
:param torch.Tensor edge_index: The edge indices. In the original formulation,
109+
the messages are aggregated from all nodes, not only from the neighbours.
109110
:return: The updated node features.
110111
:rtype: torch.Tensor
111112
"""

0 commit comments

Comments
 (0)