add return shape introduction

README.md

@@ -40,6 +40,7 @@ The Operator returns a tuple `Tuple[('feature_vector', numpy.ndarray)]` containi
 - feature_vector:
   - the embedding of the image
   - data type: `numpy.ndarray`
+  - shape: (dim,)
 
 ## Requirements
 

pytorch/model.py

@@ -32,7 +32,7 @@ class Model():
         self._model.eval()
 
     def __call__(self, img_tensor: torch.Tensor):
-        return self._model(img_tensor).detach().numpy()
+        return self._model(img_tensor).flatten().detach().numpy()
 
     def train(self):
         """