nn_model_parameter_shape

Function nn_model_parameter_shape 

Source
pub unsafe extern "C" fn nn_model_parameter_shape(
    parameter: *const NNModelParameter,
    n_dims: *mut usize,
) -> *const i32
Expand description

Returns the shape of the parameter data or NULL if no shape was defined. If n_dims is non-NULL the number of dimensions will be stored there. The shape attribute is not required for parameters but can be used either on its own or as part of defining layout of data attributes.

@public @memberof NNModelParameter @since 2.4