Unverified Commit 4f7ad258 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[doc] fix error in example (#3736)

parent 8b8fd2c0
...@@ -53,19 +53,19 @@ class EGATConv(nn.Module): ...@@ -53,19 +53,19 @@ class EGATConv(nn.Module):
>>> num_nodes, num_edges = 8, 30 >>> num_nodes, num_edges = 8, 30
>>> # generate a graph >>> # generate a graph
>>> graph = dgl.rand_graph((num_nodes,num_edges)) >>> graph = dgl.rand_graph(num_nodes,num_edges)
>>> node_feats = th.rand((num_nodes, 20)) >>> node_feats = th.rand((num_nodes, 20))
>>> edge_feats = th.rand((num_edges, 12)) >>> edge_feats = th.rand((num_edges, 12))
>>> egat = EGATConv(in_node_feats=20, >>> egat = EGATConv(in_node_feats=20,
in_edge_feats=12, ... in_edge_feats=12,
out_node_feats=15, ... out_node_feats=15,
out_edge_feats=10, ... out_edge_feats=10,
num_heads=3) ... num_heads=3)
>>> #forward pass >>> #forward pass
>>> new_node_feats, new_edge_feats = egat(graph, node_feats, edge_feats) >>> new_node_feats, new_edge_feats = egat(graph, node_feats, edge_feats)
>>> new_node_feats.shape, new_edge_feats.shape >>> new_node_feats.shape, new_edge_feats.shape
((8, 3, 12), (30, 3, 10)) torch.Size([8, 3, 15]) torch.Size([30, 3, 10])
""" """
def __init__(self, def __init__(self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment