mirror of
https://github.com/helblazer811/ManimML.git
synced 2025-05-18 03:05:23 +08:00
16 lines
555 B
Python
16 lines
555 B
Python
from manim import *
|
|
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
|
|
from manim_ml.neural_network.layers.feed_forward_to_feed_forward import (
|
|
FeedForwardToFeedForward,
|
|
)
|
|
from manim_ml.neural_network.layers.util import get_connective_layer
|
|
|
|
|
|
def test_get_connective_layer():
|
|
"""Tests get connective layer"""
|
|
input_layer = FeedForwardLayer(3)
|
|
output_layer = FeedForwardLayer(5)
|
|
connective_layer = get_connective_layer(input_layer, output_layer)
|
|
|
|
assert isinstance(connective_layer, FeedForwardToFeedForward)
|