Struct Tensor 

pub struct Tensor { /* private fields */ }

A multi-dimensional array for neural network operations.

Tensors are the primary data structure for MNN inference, holding input and output data for models.

Implementations

impl Tensor

pub unsafe fn from_ptr(ptr: *mut MNNTensor, name: Option<String>) -> Self

Create a tensor wrapper around an existing MNN tensor pointer (public version).

Safety

The pointer must be valid and remain valid for the lifetime of this tensor.

pub fn inner_mut(&mut self) -> *mut MNNTensor

Get the mutable raw pointer to the underlying MNN tensor.

pub fn as_ptr(&self) -> *const MNNTensor

Get the raw pointer to the underlying MNN tensor.

pub fn shape(&self) -> Vec<i32>

Get the shape of the tensor.

pub fn ndim(&self) -> usize

Get the number of dimensions.

pub fn dim(&self, axis: usize) -> MnnResult<i32>

Get the size of a specific dimension.

Arguments

• axis - The dimension index (0-based)

Returns

The size of the dimension, or an error if the axis is out of bounds.

pub fn dtype(&self) -> DataType

Get the data type of the tensor.

pub fn format(&self) -> DataFormat

Get the data format of the tensor.

pub fn element_count(&self) -> i32

Get the total number of elements in the tensor.

pub fn byte_size(&self) -> usize

Get the size of the tensor data in bytes.

pub fn name(&self) -> Option<&str>

Get the name of the tensor.

pub fn write<T: TensorData>(&self, data: &[T]) -> MnnResult<()>

Write data to the tensor.

Arguments

• data - The data to write

Errors

Returns an error if the data size doesn’t match.

pub fn read<T: TensorData>(&self) -> MnnResult<Vec<T>>

Read data from the tensor.

Returns

A vector containing the tensor data.

pub unsafe fn as_slice_mut<T: TensorData>(&mut self) -> MnnResult<&mut [T]>

Get a mutable reference to the tensor’s host data.

Safety

The returned slice is valid only as long as no other operations are performed on the tensor.

pub unsafe fn as_slice<T: TensorData>(&self) -> MnnResult<&[T]>

Get a reference to the tensor’s host data.

Safety

The returned slice is valid only as long as no other operations are performed on the tensor.

pub fn info(&self) -> TensorInfo

Get the tensor info.

pub fn copy_from_host(&mut self, host_tensor: &Tensor) -> MnnResult<()>

Copy data from a host tensor to this tensor (potentially on device).

Arguments

• host_tensor - The host tensor to copy from

Returns

Ok(()) on success, or an error on failure.

pub fn copy_to_host(&self, host_tensor: &mut Tensor) -> MnnResult<()>

Copy data from this tensor (potentially on device) to a host tensor.

Arguments

• host_tensor - The host tensor to copy to

Returns

Ok(()) on success, or an error on failure.

pub fn create_device( shape: &[i32], format: DataFormat, dtype: DataType, ) -> MnnResult<Tensor>

Create a device tensor with the given shape and format.

Arguments

• shape - The tensor shape
• format - The data format (NHWC, NCHW, etc.)
• dtype - The data type

Returns

A new device tensor on success, or an error on failure.

pub fn clone(&self, deep_copy: bool) -> MnnResult<Tensor>

Clone this tensor.

Arguments

• deep_copy - If true, copy data; if false, only copy metadata

Returns

A cloned tensor on success, or an error on failure.

pub fn device_id(&self) -> u64

Get the device ID for this tensor (for GPU tensors).

Returns

The device ID, or 0 if not a GPU tensor or unknown.

pub fn backend(&self) -> BackendType

Get the backend type for this tensor.

Returns

The backend type (CPU, CUDA, OpenCL, etc.).

Trait Implementations

impl Debug for Tensor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Send for Tensor

impl Sync for Tensor