mindspore.mint.masked_select

View Source On Gitee
mindspore.mint.masked_select(input, mask)[source]

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

Parameters

  • input (Tensor) – The shape of tensor is \((x_1, x_2, ..., x_R)\).

  • mask (Tensor[bool]) – The shape of tensor is \((x_1, x_2, ..., x_R)\).

Returns

A 1-D Tensor, with the same type as input.

Raises

  • TypeError – If input or mask is not a Tensor.

  • TypeError – If dtype of mask is not bool.

Supported Platforms:

Ascend

Examples

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