|
@@ -124,7 +124,7 @@ def test_decagon_layer_04(): |
|
|
|
|
|
|
|
|
seq = torch.nn.Sequential(in_layer, d_layer)
|
|
|
seq = torch.nn.Sequential(in_layer, d_layer)
|
|
|
out_d_layer = seq(None)
|
|
|
out_d_layer = seq(None)
|
|
|
out_multi_dgca = multi_dgca(in_layer(None))
|
|
|
|
|
|
|
|
|
out_multi_dgca = multi_dgca(list(in_layer(None)))
|
|
|
|
|
|
|
|
|
assert isinstance(out_d_layer, list)
|
|
|
assert isinstance(out_d_layer, list)
|
|
|
assert len(out_d_layer) == 1
|
|
|
assert len(out_d_layer) == 1
|
|
|