Skip to main content

dlarray_from_torch_f64

scirs2_numpy::dlpack

Function dlarray_from_torch_f64 

Source 
pub unsafe fn dlarray_from_torch_f64<'a>(
    tensor: *const DLTensor,
) -> Result<ArrayViewD<'a, f64>, DlpackError>
Expand description

Convert a raw DLTensor pointer (from a PyTorch DLPack capsule) to an ndarray dynamic array view of f64 elements.

§Safety

Same invariants as dlarray_from_torch_f32, but the data must be 64-bit floating-point (f64).

§Errors

Returns DlpackError if the tensor is not a 64-bit float CPU tensor.