Función torch.masked_select de PyTorch
Manual de referencia de torch de Pytorch
torch.masked_selectEs una función de PyTorch utilizada para seleccionar elementos según una máscara booleana.
Definición de la función
torch.masked_select(input, mask)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([1, 2, 3, 4, 5])
mask = torch.tensor([True, False, True, False, True])
# Seleccionar elementos según la máscara
result = torch.masked_select(x, mask)
print(result)
x = torch.tensor([1, 2, 3, 4, 5])
mask = torch.tensor([True, False, True, False, True])
# Seleccionar elementos según la máscara
result = torch.masked_select(x, mask)
print(result)
El resultado es:
tensor([1, 3, 5])
Otras extensiones