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)