add einsum support to pytorch parser

Author: JanFSchulte

hls4ml/converters/pytorch/core.py

@@ -158,7 +158,6 @@ def parse_batchnorm_layer(operation, layer_name, input_names, input_shapes, node
 
     return layer, [shape for shape in input_shapes[0]]
 
-
 @pytorch_handler('einsum')
 def parse_einsum_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config):
     assert 'einsum' in operation

test/pytest/test_pytorch_api.py

@@ -878,15 +878,13 @@ def forward(self, x):
     atol = 5.0e-2
     np.testing.assert_allclose(hls_prediction, pytorch_prediction, rtol=rtol, atol=atol)
 
-
 class EinsumOuterProduct(nn.Module):
     def __init__(self):
         super().__init__()
 
     def forward(self, x, y):
         return torch.einsum('bi,bj->bij', x, y)
 
-
 class EinsumBatchMatMul(nn.Module):
     def __init__(self):
         super().__init__()