support_to_scalar

class maze.train.trainers.common.value_transform.support_to_scalar(logits: torch.Tensor, support_range: Tuple[int, int])

Convert support vector to scalar by probability weighted interpolation.

Parameters:

• logits – Logits fed into a softmax to get probabilities in range [0, 1].

• support_range – Tuple holding the lower and upper bound of the supported value range.

Returns:

Tensor of converted scalars.