Function cast

pub fn cast<R: JitRuntime, EI: JitElement, EO: JitElement>(
    input: JitTensor<R>,
) -> JitTensor<R>

Cast a tensor to the given element type.

Note: When input element is semantically a boolean, prefer bool_cast function.