Fonction torch.masked_select de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.masked_selectC'est une fonction de PyTorch utilisée pour sélectionner des éléments en fonction d'un masque booléen.

Définition de fonction

torch.masked_select(input, mask)

Exemple d'utilisation

Exemple

import torch

x = torch.tensor([1, 2, 3, 4, 5])
mask = torch.tensor([True, False, True, False, True])

# Sélectionner les éléments en fonction du masque
result = torch.masked_select(x, mask)

print(result)

Le résultat de sortie est:

tensor([1, 3, 5])

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions