From ec6356605a506092c788ca8b92b5b12c25142b1b Mon Sep 17 00:00:00 2001 From: n-shevko Date: Thu, 6 Mar 2025 15:25:33 -0500 Subject: [PATCH] Note about performance --- docs/source/guide/guide_part_i.rst | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docs/source/guide/guide_part_i.rst b/docs/source/guide/guide_part_i.rst index a66d18fb..a10800ca 100644 --- a/docs/source/guide/guide_part_i.rst +++ b/docs/source/guide/guide_part_i.rst @@ -193,6 +193,9 @@ Example 2: Batch dimension specified via batch_size parameter (no batch axis in weights = Weight(name='weight_feature', value=torch.rand(100, 1000), sparse=True, batch_size=2) bias = Bias(name='bias_feature', value=torch.rand(100, 1000), sparse=True, batch_size=2) + +Note that subtraction and addition operations for sparse tensors are 90 times slower than those for dense tensors (tested on RTX 3060) + Specifying monitors *******************