add xlib.avecl

xlib/avecl/_internal/HTensor.py

@@ -0,0 +1,22 @@
+from typing import List, Union
+
+from .Tensor import *
+
+
+class HTensor:
+    """
+    Helper functions for Tensor
+    """
+
+    @staticmethod
+    def all_same_device( tensor_or_list : Union[Tensor, List[Tensor] ] ) -> bool:
+        """
+        check if all tensors in a list use the same device
+        """
+        if not isinstance(tensor_or_list, (list,tuple)):
+            tensor_or_list = (tensor_or_list,)
+
+        device = tensor_or_list[0].get_device()
+        return all( device == tensor.get_device() for tensor in tensor_or_list )
+
+__all__ = ['HTensor']