I v súčasnosti majú dve Baterky Ansi, p
a x
, ktoré obe majú tvar (batch_size, input_size)
.
Ja by som chcel vypočítať Bernoulli denník likelihoods pre dané dáta, a vrátenie tensor veľkosť (batch_size)
Tu je príklad, čo by som chcel robiť: Mám vzorec pre denník likelihoods z Bernoulli Náhodné premenné:
\sum_i^d x_{i} ln(p_i) + (1-x_i) ln (1-p_i)
Povedať, že som p
Tensor:
[[0.6 0.4 0], [0.33 0.34 0.33]]
A povedať, že mám x
tensor pre binárne vstupy na základe týchto pravdepodobností:
[[1 1 0], [0 1 1]]
A chcem vypočítať log likelihood pre každú vzorku, ktorá by mala za následok:
[[ln(0.6)+ln(0.4)], [ln(0.67)+ln(0.34)+ln(0.33)]]
Bolo by možné urobiť výpočet bez použitia pre slučky?
Viem, že by som mohol použiť torch.sum(axis=1)
robiť konečného súčtu medzi denníky, ale je to možné urobiť Bernoulli log-likelihood výpočtu bez použitia pre slučky? alebo použitie na väčšine 1 pre slučky? Snažím sa vectorize tejto operácii, rovnako ako je to možné. Ja by som prisahal by sme mohli používať Latexové pre rovníc pred, niečo zmeniť, alebo je to ďalší stránky?