mindspore.Tensor.masked_select

Tensor.masked_select(mask)[source]

Returns a new 1-D Tensor which indexes the self tensor according to the boolean mask. The shapes of the mask tensor and the self tensor don’t need to match, but they must be broadcastable.

Parameters: mask (Tensor[bool]) – The boolean Tensor.
Returns: A 1-D Tensor, with the same type as self.
Raises: TypeError – If mask is not a bool Tensor.

Supported Platforms:: Ascend GPU CPU

Examples

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([1, 2, 3, 4]), mindspore.int64)
>>> mask = Tensor(np.array([1, 0, 1, 0]), mindspore.bool_)
>>> output = x.masked_select(mask)
>>> print(output)
[1 3]