"vscode:/vscode.git/clone" did not exist on "e96860d60be171e0802cdbd180ca976c1afd2b50"
Unverified Commit 1bd36886 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Tutorial] Add a small test for GAT exercise (#5644)


Co-authored-by: default avatarHongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
parent 24b7468f
...@@ -1307,6 +1307,21 @@ ...@@ -1307,6 +1307,21 @@
}, },
"execution_count": null, "execution_count": null,
"outputs": [] "outputs": []
},
{
"cell_type": "code",
"source": [
"# Test:\n",
"# Let's use the symmetric A created above.\n",
"X = torch.randn(6, 20)\n",
"module = SimplifiedGAT(20, 10)\n",
"Y = module(A, X)"
],
"metadata": {
"id": "qjcXiidYCqGK"
},
"execution_count": null,
"outputs": []
} }
] ]
} }
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