Skip to main content

Module tensor

Module tensor 

Source
Expand description

MNN::Tensor related items

Modules§

list

Structs§

AnyTensorRef
This is a zero sized type
Device
A tensor that is owned by the device / gpu platform
Host
A tensor that is owned by the cpu / host platform
Owned
A tensor that is owned by the host / device platform
Tensor
A generic tensor that can of host / device / owned / borrowed
TensorRef
This is modeled as a reference to a tensor, you should never be able to dereference this. At any point you should only have &TensorRef or &mut TensorRef, never an owned TensorRef.
TensorShape
A tensor shape
View
A reference to a any tensor

Enums§

DimensionType
The type of the tensor dimension If you are manually specifying the shapes then this doesn’t really matter N -> Batch size C -> Channel H -> Height W -> Width

Traits§

AsTensorShape
A trait to convert any array-like type to a tensor shape
MutableTensorType
A trait to represent a mutable tensor type
TensorMachine
A trait to represent the device type of a tensor
TensorType
A trait to represent the type of a tensor

Functions§

from_raw_parts⚠
Construct a tensor reference from a raw pointer to an MNN tensor
from_raw_parts_mut⚠
Construct a mutable tensor reference from a raw pointer to an MNN tensor

Type Aliases§

TensorOwned
A tensor that can be owned by the host or device
TensorView
A tensor that can be borrowed from the host or device
TensorViewMut
A tensor that can be borrowed mutably from the host or device