rbnet.util.log_normalize

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

Normalise tensor t in log representation by computing

using PyTorch logsumexp.

Parameters:

  • t

  • args – positional arguments passed on to logsumexp

  • kwargs – key-word arguments passed on to logsumexp

Returns:

normalised tensor