Skip to contents

Per-row mixture log density (in standardized theta space), as a torch tensor. theta: (b, p) tensor, x: (b, q) tensor.

Usage

mdn_log_prob_tensor(net, theta, x)