Struct Tensor 

pub struct Tensor(/* private fields */);

A multi-dimensional array for machine learning.

C++ equivalent: MTL::Tensor

Implementations

impl Tensor

pub unsafe fn from_raw(ptr: *mut c_void) -> Option<Self>

Create from a raw pointer.

Safety

The pointer must be a valid Metal Tensor.

pub fn as_raw(&self) -> *mut c_void

Get the raw pointer.

pub fn buffer(&self) -> Option<Buffer>

Get the backing buffer.

C++ equivalent: Buffer* buffer() const

pub fn buffer_offset(&self) -> UInteger

Get the buffer offset.

C++ equivalent: NS::UInteger bufferOffset() const

pub fn data_type(&self) -> TensorDataType

Get the data type.

C++ equivalent: TensorDataType dataType() const

pub fn dimensions(&self) -> Option<TensorExtents>

Get the dimensions.

C++ equivalent: TensorExtents* dimensions() const

pub fn strides(&self) -> Option<TensorExtents>

Get the strides.

C++ equivalent: TensorExtents* strides() const

pub fn usage(&self) -> TensorUsage

Get the usage.

C++ equivalent: TensorUsage usage() const

pub fn gpu_resource_id(&self) -> ResourceID

Get the GPU resource ID.

C++ equivalent: ResourceID gpuResourceID() const

pub fn get_bytes( &self, bytes: *mut c_void, strides: &TensorExtents, slice_origin: &TensorExtents, slice_dimensions: &TensorExtents, )

Get bytes from the tensor.

C++ equivalent: void getBytes(void*, const TensorExtents*, const TensorExtents*, const TensorExtents*)

pub fn replace_slice_origin( &self, slice_origin: &TensorExtents, slice_dimensions: &TensorExtents, bytes: *const c_void, strides: &TensorExtents, )

Replace a slice in the tensor.

C++ equivalent: void replaceSliceOrigin(const TensorExtents*, const TensorExtents*, const void*, const TensorExtents*)

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Drop for Tensor

fn drop(&mut self)

Executes the destructor for this type.

impl Referencing for Tensor

fn as_ptr(&self) -> *const c_void

Get the raw pointer to the Objective-C object.

fn as_mut_ptr(&self) -> *mut c_void

Get the raw mutable pointer to the Objective-C object.

fn retain(&self) -> Self

where Self: Clone,

Retain the object, incrementing its reference count.

fn release(&self)

Release the object, decrementing its reference count.

fn autorelease(&self) -> Self

where Self: Clone,

Autorelease the object.

fn retain_count(&self) -> usize

Get the retain count of the object.

impl Send for Tensor

impl Sync for Tensor