torch.distributions.utils.clamp_probs

torch.distributions.utils.clamp_probs(probs) [исходный код]

Ограничивает значения вероятностей открытым интервалом (0, 1).

Значения вероятностей будут ограничены диапазоном от eps до 1 - eps, а eps будет наименьшим представимым положительным числом для типа входных данных.

Параметры:

probs (Tensor) – Тензор вероятностей.

Возвращает:

Ограниченные значения вероятностей.

Тип возвращаемого значения:

Tensor

Примеры

>>> probs = torch.tensor([0.0, 0.5, 1.0])
>>> clamp_probs(probs)
tensor([1.1921e-07, 5.0000e-01, 1.0000e+00])

>>> probs = torch.tensor([0.0, 0.5, 1.0], dtype=torch.float64)
>>> clamp_probs(probs)
tensor([2.2204e-16, 5.0000e-01, 1.0000e+00], dtype=torch.float64)