Struct Tensor

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

A generic tensor that can of host / device / owned / borrowed

Implementations

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

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

Get’s a reference to an owned host tensor

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

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

Get’s a reference to an owned device tensor

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

Get the device id of the tensor

pub fn shape(&self) -> TensorShape

Get the shape of the tensor

pub fn dimensions(&self) -> usize

Get the dimensions of the tensor

pub fn width(&self) -> u32

Get the width of the tensor

pub fn height(&self) -> u32

Get the height of the tensor

pub fn channel(&self) -> u32

Get the channel size of the tensor

pub fn batch(&self) -> u32

Get the batch size of the tensor

pub fn size(&self) -> usize

Get the size of the tensor when counted by bytes

pub fn element_size(&self) -> usize

Get the size of the tensor when counted by elements

pub fn print_shape(&self)

Print the shape of the tensor

pub fn print(&self)

Print the tensor

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

Get the dimension type of the tensor

pub fn get_type(&self) -> halide_type_t

Get the data type of the tensor

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

Check if the tensor is of the specified data type

pub unsafe fn into_raw(self) -> RawTensor<'static>

Safety

This is very unsafe do not use this unless you know what you are doing

impl<T: MutableTensorType> Tensor<T>

where T::H: HalideType,

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

where T::H: Copy,

Fill the tensor with the specified value

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]

Get the host memory slice of the tensor

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

Get the mutable host memory slice of the tensor

impl<T: DeviceTensorType> Tensor<T>

where T::H: HalideType,

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

Try to wait for the device tensor to finish processing

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

Create a host tensor from the device tensor with same dimensions and data type and optionally copy the data from the device tensor

impl<T: OwnedTensorType> Tensor<T>

where T::H: HalideType,

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

Create a new tensor with the specified shape and dimension type

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

where T::H: HalideType,

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

Try to create a ref tensor from any array-like type

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

Try to create a mutable ref tensor from any array-like type

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.