Masked_select
Masked_select
torch_masked_select(self, mask)
self |
(Tensor) the input tensor. |
mask |
(BoolTensor) the tensor containing the binary mask to index with |
Returns a new 1-D tensor which indexes the input
tensor according to
the boolean mask mask
which is a BoolTensor
.
The shapes of the mask
tensor and the input
tensor don't need
to match, but they must be broadcastable .
The returned tensor does not use the same storage as the original tensor
if (torch_is_installed()) { x = torch_randn(c(3, 4)) x mask = x$ge(0.5) mask torch_masked_select(x, mask) }
Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.