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

Added working convolutional layer.

Browse Source
This commit is contained in:
Alec Helbling
2022-04-29 14:36:14 -04:00
parent 9d6edbc6d7
commit 8d45c13f57
7 changed files with 278 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
manim_ml/neural_network/layers/util.py
View File

@ -1,4 +1,6 @@
from manim import *
from manim_ml.neural_network.layers.parent_layers import BlankConnective
from ..layers import connective_layers_list
def get_connective_layer(input_layer, output_layer):
@ -14,7 +16,10 @@ def get_connective_layer(input_layer, output_layer):
connective_layer = connective_layer_class(input_layer, output_layer)
if connective_layer is None:
connective_layer = BlankConnective(input_layer, output_layer)
"""
raise Exception(f"Unrecognized class pair {input_layer.__class__.__name__}" + \
f" and {output_layer.__class__.__name__}")
"""
return connective_layer