mirror of
https://github.com/helblazer811/ManimML.git
synced 2025-05-26 02:05:18 +08:00
63427be139
for arguments to be passed through to each neural network layer when running a neural network forward pass.
26 lines
790 B
Python
26 lines
790 B
Python
|
|
from manim import *
|
|
from manim_ml.neural_network.layers.parent_layers import VGroupNeuralNetworkLayer
|
|
|
|
class ConvolutionalLayer(VGroupNeuralNetworkLayer):
|
|
"""Handles rendering a convolutional layer for a nn"""
|
|
|
|
def __init__(self, num_filters, filter_width, **kwargs):
|
|
super(VGroupNeuralNetworkLayer, self).__init__(**kwargs)
|
|
self.num_filters = num_filters
|
|
self.filter_width = filter_width
|
|
|
|
self._construct_neural_network_layer()
|
|
|
|
def _construct_neural_network_layer(self):
|
|
"""Creates the neural network layer"""
|
|
pass
|
|
|
|
def make_forward_pass_animation(self, layer_args={}, **kwargs):
|
|
# make highlight animation
|
|
return None
|
|
|
|
@override_animation(Create)
|
|
def _create_override(self, **kwargs):
|
|
pass
|