PyTorch torch.where-Funktion
Pytorch torch-Referenzhandbuch
torch.whereEs ist eine Funktion in PyTorch, die verwendet wird, um Elemente basierend auf Bedingungen zurückzugeben.
Funktionsdefinition
torch.where(condition, input, other)
Anwendungsbeispiel
Beispiel
import torch
Bedingung = torch.tensor([[True, False], [False, True]])
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[10, 20], [30, 40]])
Ergebnis = torch.where(Bedingung, x, y)
print(result)
Bedingung = torch.tensor([[True, False], [False, True]])
x = torch.tensor([[1, 2], [3, 4]])
y = torch.tensor([[10, 20], [30, 40]])
Ergebnis = torch.where(Bedingung, x, y)
print(result)
Das Ausgabeergebnis ist:
tensor([[ 1, 20],
[30, 4]])
Weitere Erweiterungen