masked_select¶
dragon.vm.torch.
masked_select
(
input,
mask,
out=None
)[source]¶Select the input elements where mask is true.
- Parameters:
- input (dragon.vm.torch.Tensor) – The input tensor.
- mask (dragon.vm.torch.Tensor) – The mask for selecting.
- out (dragon.vm.torch.Tensor, optional) – The output tensor.
- Returns:
dragon.vm.torch.Tensor – The output tensor.