Struct mnn::tensor::Tensor

pub struct Tensor<T: TensorType> { /* private fields */ }

Implementations

impl<H: HalideType> Tensor<Host<H>>

pub fn as_ref(&self) -> Tensor<Ref<'_, Host<H>>>

impl<H: HalideType> Tensor<Device<H>>

pub fn as_ref(&self) -> Tensor<Ref<'_, Device<H>>>

impl<T: TensorType> Tensor<T>

where T::H: HalideType,

pub unsafe fn from_ptr(tensor: *mut Tensor) -> Self

This function constructs a Tensor type from a raw pointer

Safety

Since this constructs a Tensor from a raw pointer we have no way to guarantee that it’s a valid tensor or it’s lifetime

pub fn copy_from_host_tensor( &mut self, tensor: &Tensor<Host<T::H>>, ) -> Result<()>

Copies the data from a host tensor to the self tensor

pub fn copy_to_host_tensor(&self, tensor: &mut Tensor<Host<T::H>>) -> Result<()>

Copies the data from the self tensor to a host tensor

pub fn device_id(&self) -> u64

pub fn shape(&self) -> TensorShape

Get the shape of the tensor

pub fn dimensions(&self) -> usize

pub fn width(&self) -> u32

pub fn height(&self) -> u32

pub fn channel(&self) -> u32

pub fn batch(&self) -> u32

pub fn size(&self) -> usize

pub fn element_size(&self) -> usize

pub fn print_shape(&self)

pub fn print(&self)

pub fn is_dynamic_unsized(&self) -> bool

Check if the tensor is dynamic and needs resizing

pub unsafe fn halide_buffer(&self) -> *const halide_buffer_t

DO not use this function directly

Safety

This is just provided as a 1:1 compat mostly for possible later use

pub unsafe fn halide_buffer_mut(&self) -> *mut halide_buffer_t

Do not use this function directly

Safety

This is just provided as a 1:1 compat mostly for possible later use

pub fn get_dimension_type(&self) -> DimensionType

pub fn get_type(&self) -> halide_type_t

pub fn is_type_of<H: HalideType>(&self) -> bool

impl<T: MutableTensorType> Tensor<T>

where T::H: HalideType,

pub fn fill(&mut self, value: T::H)

where T::H: Copy,

impl<T: HostTensorType> Tensor<T>

where T::H: HalideType,

pub fn try_host(&self) -> Result<&[T::H]>

Try to map the device tensor to the host memory and get the slice

pub fn try_host_mut(&mut self) -> Result<&mut [T::H]>

Try to map the device tensor to the host memory and get the mutable slice

pub fn host(&self) -> &[T::H]

pub fn host_mut(&mut self) -> &mut [T::H]

impl<T: DeviceTensorType> Tensor<T>

where T::H: HalideType,

pub fn wait(&self, map_type: MapType, finish: bool)

pub fn create_host_tensor_from_device( &self, copy_data: bool, ) -> Tensor<Host<T::H>>

impl<T: OwnedTensorType> Tensor<T>

where T::H: HalideType,

pub fn new(shape: impl AsTensorShape, dm_type: DimensionType) -> Self

impl<T: HostTensorType + RefTensorType> Tensor<T>

where T::H: HalideType,

pub fn borrowed(shape: impl AsTensorShape, input: impl AsRef<[T::H]>) -> Self

pub fn borrowed_mut( shape: impl AsTensorShape, input: impl AsMut<[T::H]>, ) -> Self

Trait Implementations

impl<T: OwnedTensorType> Clone for Tensor<T>

where T::H: HalideType,

fn clone(&self) -> Tensor<T>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: TensorType> Drop for Tensor<T>

fn drop(&mut self)

Executes the destructor for this type.