fix transformers w class ut and diffusers omni (#2092)

Author: lvyufeng

mindnlp/core/_tensor.py

@@ -711,6 +711,27 @@ def __contains__(self, item):
     Tensor.var = ops.var
     StubTensor.var = ops.var
 
+    Tensor.logsumexp = ops.logsumexp
+    StubTensor.logsumexp = ops.logsumexp
+
+    def __bool__(self):
+        if self.ndim > 0:
+            return True
+        return bool(self._item())
+
+    Tensor.__bool__ = __bool__
+    StubTensor.__bool__ = __bool__
+
+    def __iter__(self):
+        if self.ndim == 0:
+            yield self
+        else:
+            for i in range(len(self)):
+                yield self[i]
+
+    Tensor.__iter__ = __iter__
+    StubTensor.__iter__ = __iter__
+
 def _rebuild_from_type_v2(func, new_type, args, state):
     ret = func(*args)
     return ret

mindnlp/core/ops/other.py

@@ -1044,6 +1044,9 @@ def cartesian_prod(*tensors):
 def detach(input):
     return ops.stop_gradient(input)
 
+def cosine_similarity(*args, **kwargs):
+    return core.nn.functional.cosine_similarity(*args, **kwargs)
+
 __all__ = [
     "bincount",
     "broadcast_shapes",
@@ -1081,5 +1084,6 @@ def detach(input):
     "histc",
     "view_as_complex",
     "view_as_real",
-    "detach"
+    "detach",
+    "cosine_similarity"
 ]