tensor_clamp

class maze.distributions.utils.tensor_clamp(x: torch.Tensor, t_min: torch.Tensor, t_max: torch.Tensor)

Clamping with tensor and broadcast support.

Parameters:

• x – the tensor to clamp.

• t_min – the minimum values.

• t_max – the maximum values.

Returns:

the clamped tensor.