Exemplo n.º 1
0
def test_nested_sequential_network():
    current_node = InputNode("foo", shape=(3, 4, 5))
    for name in map(str, range(10)):
        current_node = SequentialNode("sequential" + name,
                                      [current_node,
                                       IdentityNode("identity" + name)])
    network = current_node.network()
    fn = network.function(["foo"], ["sequential9"])
    x = np.random.rand(3, 4, 5).astype(fX)
    assert np.allclose(fn(x), x)
Exemplo n.º 2
0
def test_identity_network():
    input_node = InputNode("foo", shape=(3, 4, 5))
    network = input_node.network()
    fn = network.function(["foo"], ["foo"])
    x = np.random.rand(3, 4, 5).astype(fX)
    assert np.allclose(fn(x), x)