Skip to content

Commit 2d05f8b

Browse files
committed
correct the docstring of get_affine_components to respect PEP-257 convention
1 parent 36a6bcf commit 2d05f8b

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/decomon/backward_layers/backward_layers.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,6 @@ def __init__(
343343

344344
self.frozen_weights = False
345345

346-
347346
def get_config(self):
348347
config = super().get_config()
349348
config.update(
@@ -356,9 +355,11 @@ def get_config(self):
356355
return config
357356

358357
def get_affine_components(self, x):
359-
"""Conv is a linear operator but its affine component is implicit
360-
we use im2col and extract_patches to express the affine matrix
361-
Note that this matrix is Toeplitz
358+
"""Express the implicit affine matrix of the convolution layer.
359+
360+
Conv is a linear operator but its affine component is implicit
361+
we use im2col and extract_patches to express the affine matrix
362+
Note that this matrix is Toeplitz
362363
363364
Args:
364365
x: list of input tensors

0 commit comments

Comments
 (0)