Maskierte Aggregationen in Pytorch ⇐ Python

[Anonymous]

Gegebene Daten und Masken-Tensoren Gibt es einen Pytorch-Weg, um maskierte Datenaggregationen (Mittelwert, max, min usw.) zu erhalten. < /p>

Code: Select all

x = torch.tensor([
[1, 2, -1, -1],
[10, 20, 30, -1]
])

mask = torch.tensor([
[True, True, False, False],
[True, True, True, False]
])
< /code>
Um einen maskierten Mittel zu berechnen, kann ich Folgendes tun, aber gibt es ein integriertes oder häufig verwendetes Pytorch-Paket, um dies zu tun? < /p>
n_mask = torch.sum(mask, axis=1)
x_mean = torch.sum(x * mask, axis=1) / n_mask

print(x_mean)
< /code>
> tensor([ 1.50, 20.00])