torch_can_cast: Can_cast

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Can_cast

Description

Can_cast

Usage

torch_can_cast(from, to)

Arguments

from	(dtype) The original torch_dtype.
to	(dtype) The target torch_dtype.

can_cast(from, to) -> bool

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

Examples

if (torch_is_installed()) {

torch_can_cast(torch_double(), torch_float())
torch_can_cast(torch_float(), torch_int())
}

torch documentation built on June 7, 2023, 6:19 p.m.