tkt

byt3berry · Feb 12, 2022 · bd47948 · bd47948
1 parent d014213
commit bd47948
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Classes/PoolingLayer.py b/Classes/PoolingLayer.py
@@ -33,7 +33,7 @@ def feedForward(self, inputs: list[Matrice]) -> list[Matrice]:
 
         return self.outputs
 
-    def pooling(self, input_: Matrice):
+    def pooling(self, input_: Matrice) -> Matrice:
         output = Matrice.vide(input_.getRows() // self.filterDim, input_.getColumns() // self.filterDim)
 
         for i in range(output.getRows()):