From e6929631c9ab529dee65e74404f763ba666c4559 Mon Sep 17 00:00:00 2001 From: Lionel Date: Fri, 10 Jan 2025 10:46:28 +0100 Subject: [PATCH] small change in docstring --- lightly/models/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lightly/models/utils.py b/lightly/models/utils.py index 3a29d0c43..9ac703182 100644 --- a/lightly/models/utils.py +++ b/lightly/models/utils.py @@ -40,9 +40,9 @@ def pool_masked( integer indices. num_cls: The number of classes in the possible masks. - Returns: - A tensor of shape :math:`(B, C, N)` or :math:`(C, N)` where :math:`N` - corresponds to `num_cls`. + Returns: + A tensor of shape :math:`(B, C, N)` or :math:`(C, N)` where :math:`N` + corresponds to `num_cls`. """ if source.dim() == 3: return _mask_reduce(source, mask, reduce, num_cls)