opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-05-31 06:08:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Finshed adding construct_layer methods for refactor.

Browse Source
This commit is contained in:
Alec Helbling
2023-01-15 16:52:22 +09:00
parent 42b6e37b16
commit 99dbda915b
18 changed files with 60 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
manim_ml/neural_network/layers/max_pooling_2d.py
View File

@ -26,14 +26,12 @@ class MaxPooling2DLayer(VGroupNeuralNetworkLayer, ThreeDLayer):
self.kernel_size = kernel_size
self.stride = stride
self.cell_highlight_color = cell_highlight_color
def construct_layer(self, input_layer: 'NeuralNetworkLayer', output_layer: 'NeuralNetworkLayer', **kwargs):
# Make the output feature maps
feature_maps = self._make_output_feature_maps()
self.add(feature_maps)
def construct_layer(self, input_layer, output_layer):
"""Constructs the layer in the context of adjacent layers"""
pass
def _make_output_feature_maps(self):
"""Makes a set of output feature maps"""
# Compute the size of the feature maps