masked_select¶

paddle. masked_select ( x, mask, name=None ) [source]

Returns a new 1-D tensor which indexes the input tensor according to the mask which is a tensor with data type of bool.

Parameters

x (Tensor) – The input Tensor, the data type can be int32, int64, uint16, float16, float32, float64.
mask (Tensor) – The Tensor containing the binary mask to index with, it’s data type is bool.
name (str, optional) – For details, please refer to Name. Generally, no setting is required. Default: None.

Returns

A 1-D Tensor which is the same data type as x.

Examples

           >>> import paddle

>>> x = paddle.to_tensor([[1.0, 2.0, 3.0, 4.0],
...                       [5.0, 6.0, 7.0, 8.0],
...                       [9.0, 10.0, 11.0, 12.0]])
>>> mask = paddle.to_tensor([[True, False, False, False],
...                          [True, True, False, False],
...                          [True, False, False, False]])
>>> out = paddle.masked_select(x, mask)
>>> print(out.numpy())
[1. 5. 6. 9.]

          