Module tensor

Source
Expand description

MNN::Tensor related items

Structs§

Device
A tensor that is owned by the device / gpu platform
Host
A tensor that is owned by the cpu / host platform
RawTensor
A raw tensor type that doesn’t have any guarantees and will be unconditionally dropped
Ref
A reference to a any tensor
RefMut
A mutable reference to a any tensor
Tensor
A generic tensor that can of host / device / owned / borrowed
TensorShape
A tensor shape

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
DeviceTensorType
A tensor that is allocated in the device / gpu platform
HostTensorType
A tensor that is allocated in the cpu / host platform
MutableTensorType
A tensor that is mutable
OwnedTensorType
A tensor that is owned
RefTensorType
A tensor that is borrowed
TensorType
A trait to represent the type of a tensor