mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-04 19:44:23 +08:00
[Tutorial] Add a small test for GAT exercise (#5644)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
This commit is contained in:
@@ -1307,6 +1307,21 @@
|
||||
},
|
||||
"execution_count": null,
|
||||
"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": []
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user