File tree 1 file changed +5
-4
lines changed
src/decomon/backward_layers
1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -343,7 +343,6 @@ def __init__(
343
343
344
344
self .frozen_weights = False
345
345
346
-
347
346
def get_config (self ):
348
347
config = super ().get_config ()
349
348
config .update (
@@ -356,9 +355,11 @@ def get_config(self):
356
355
return config
357
356
358
357
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
362
363
363
364
Args:
364
365
x: list of input tensors
You can’t perform that action at this time.
0 commit comments