Skip to main content

create_ndarray

Function create_ndarray 

Source
pub unsafe extern "C" fn create_ndarray(
    shape: *const usize,
    shape_count: usize,
    dtype: DLDataType,
    array: *mut mts_array_t,
) -> mts_status_t
Expand description

Callback function to create a new mts_array_t when loading TensorMap or TensorBlock from a buffer or a file.

This is an implementation of mts_create_array_callback_t that creates a new ndarray::ArrayD as the backing array for the mts_array_t. It is used by default in load_buffer and load_block_buffer.