From 3b33028a80aa36866e7dbec4afed67188ccc8446 Mon Sep 17 00:00:00 2001 From: lvyufeng Date: Mon, 21 Jul 2025 14:30:11 +0800 Subject: [PATCH] fix h class ut --- mindnlp/core/_tensor.py | 9 +++++++++ mindnlp/core/ops/array.py | 4 +++- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/mindnlp/core/_tensor.py b/mindnlp/core/_tensor.py index 66388b39c..b2dbe556c 100644 --- a/mindnlp/core/_tensor.py +++ b/mindnlp/core/_tensor.py @@ -570,6 +570,15 @@ def sort(self, dim=-1, descending=False): Tensor.scatter_ = ops.inplace_scatter StubTensor.scatter_ = ops.inplace_scatter + def __contains__(self, item): + return ops.equal(self, item).any() + + Tensor.__contains__ = __contains__ + StubTensor.__contains__ = __contains__ + + Tensor.tile = ops.tile + StubTensor.tile = ops.tile + def _rebuild_from_type_v2(func, new_type, args, state): ret = func(*args) return ret \ No newline at end of file diff --git a/mindnlp/core/ops/array.py b/mindnlp/core/ops/array.py index 2e981fc43..bd01b32c2 100644 --- a/mindnlp/core/ops/array.py +++ b/mindnlp/core/ops/array.py @@ -344,7 +344,9 @@ def tensor_split(input, indices_or_sections, dim=0): # tile has_tile = hasattr(mindspore.mint, 'tile') -def tile(input, dims): +def tile(input, *dims): + if isinstance(dims[0], tuple): + dims = dims[0] if use_pyboost() and has_tile: return mindspore.mint.tile(input, dims) return ops.tile(input, dims)