(
t: Tensor,
continuous_range = Tuple[float, float],
num_discrete: int = 128
)
| 21 | |
| 22 | |
| 23 | def undiscretize( |
| 24 | t: Tensor, |
| 25 | continuous_range = Tuple[float, float], |
| 26 | num_discrete: int = 128 |
| 27 | ) -> Tensor: |
| 28 | lo, hi = continuous_range |
| 29 | assert hi > lo |
| 30 | t = t.float() |
| 31 | t += 0.5 |
| 32 | t /= num_discrete # cube normalize |
| 33 | return t * (hi - lo) + lo |
| 34 | |
| 35 | |
| 36 |