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.