Tensor Attributes

Tensor Views

Complex Numbers

Quantization

Distributed RPC Framework

Type Info

Named Tensors

Named Tensors operator coverage

API

torch / torch

torch.can_cast¶

torch.can_cast(from, to) → bool¶

Determines if a type conversion is allowed under PyTorch casting rules described in the type promotion documentation.

Parameters

from (dpython:type) – The original torch.dtype.
to (dpython:type) – The target torch.dtype.

Example:

>>> torch.can_cast(torch.double, torch.float)
True
>>> torch.can_cast(torch.float, torch.int)
False

此页内容是否对您有帮助