MCPcopy
hub / github.com/dmlc/dgl / message_func

Function message_func

tests/python/common/function/test_basics.py:15–18  ·  view source on GitHub ↗
(edges)

Source from the content-addressed store, hash-verified

13
14
15def message_func(edges):
16 assert F.ndim(edges.src["h"]) == 2
17 assert F.shape(edges.src["h"])[1] == D
18 return {"m": edges.src["h"]}
19
20
21def reduce_func(nodes):

Callers

nothing calls this directly

Calls 1

shapeMethod · 0.45

Tested by

no test coverage detected