|
@@ -166,7 +166,7 @@ def test_decode_layer_05(): |
|
|
# print(rel_pred.edges_neg.train)
|
|
|
# print(rel_pred.edges_neg.train)
|
|
|
|
|
|
|
|
|
repr_in = in_layer(None)
|
|
|
repr_in = in_layer(None)
|
|
|
assert isinstance(repr_in, list)
|
|
|
|
|
|
|
|
|
assert isinstance(repr_in, torch.nn.ParameterList)
|
|
|
assert len(repr_in) == 1
|
|
|
assert len(repr_in) == 1
|
|
|
assert isinstance(repr_in[0], torch.Tensor)
|
|
|
assert isinstance(repr_in[0], torch.Tensor)
|
|
|
assert torch.all(repr_in[0].to_dense() == torch.eye(10))
|
|
|
assert torch.all(repr_in[0].to_dense() == torch.eye(10))
|
|
|