torch.can_cast
-
torch.can_cast(from, to) → bool -
Определяет, разрешена ли преобразование типа по правилам приведения типов PyTorch, описанным в документации.
- Параметры
-
-
from (тип) – Исходный
torch.dtype. -
to (тип) – Целевой
torch.dtype.
-
from (тип) – Исходный
Пример:
>>> torch.can_cast(torch.double, torch.float) True >>> torch.can_cast(torch.float, torch.int) False
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.can_cast.html
