From cc7a0cfdc78d53a797cc5f765fa2a62da09ae1f5 Mon Sep 17 00:00:00 2001 From: Akihiro Nitta Date: Mon, 22 Jul 2024 21:37:30 +0000 Subject: [PATCH] update --- pyg_lib/ops/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pyg_lib/ops/__init__.py b/pyg_lib/ops/__init__.py index e9b06945..bdf8efd5 100644 --- a/pyg_lib/ops/__init__.py +++ b/pyg_lib/ops/__init__.py @@ -104,8 +104,7 @@ def grouped_matmul( r"""Performs dense-dense matrix multiplication according to groups, utilizing dedicated kernels that effectively parallelize over groups. - .. code-block:: python - + Example: inputs = [torch.randn(5, 16), torch.randn(3, 32)] others = [torch.randn(16, 32), torch.randn(32, 64)]