rbnet.util.log_normalize

rbnet.util.log_normalize(t, *args, **kwargs)[source]

Normalise tensor t in log representation by computing t - \log \sum \exp t using PyTorch logsumexp.

Parameters:

  • t

  • args – positional arguments passed on to logsumexp

  • kwargs – key-word arguments passed on to logsumexp

Returns:

normalised tensor