nn_tensor_slice

Function nn_tensor_slice 

Source
pub unsafe extern "C" fn nn_tensor_slice(
    output: *mut NNTensor,
    input: *mut NNTensor,
    n_axes: i32,
    axes: *const i32,
    head: *const i32,
    tail: *const i32,
) -> NNError
Expand description

nn_tensor_slice copies a slice of the tensor into output. For a version which supports strides see @ref nn_tensor_strided_slice.

The axes, head, and tail must be of length n_axes or NULL. Calling slice with axes==NULL will ignore head/tail and is effectively @ref nn_tensor_copy.

When head is NULL all axes are assumed to start at 0. When tail is NULL all axes are assumed to end at (len(axis) - head) for the given axis.

@public @memberof NNTensor @since 2.0