Fonction torch.where de PyTorch

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

torch.whereC'est une fonction de PyTorch utilisée pour retourner des éléments selon des conditions.

Définition de fonction

torch.where(condition, input, other)

Exemple d'utilisation

Exemple

import torch

condition = torch.tensor([[True, False], [False, True]])
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[10, 20], [30, 40]])

result = torch.where(condition, x, y)

print(result)

Le résultat de sortie est:

tensor([[ 1, 20],
        [30,  4]])

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

Autres extensions