torch.fx.experimental.symbolic_shapes.guard_scalar

torch.fx.experimental.symbolic_shapes.guard_scalar(a) [исходный код]

Проверяет скалярное значение, которое может быть символьным или конкретным логическим значением, целым числом или числом с плавающей точкой.

Эта функция вызывает соответствующую функцию проверки в зависимости от типа входного значения.

Параметры:

a (SymBool | SymInt | SymFloat | int | bool | float) – Символьное или конкретное скалярное значение (bool, int или float)

Возвращает:

Конкретное значение после проверки

Исключения:

AssertionError – Если входное значение не относится к распознанному скалярному типу

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

bool | int | float