Struct TchTensor

pub struct TchTensor {
    pub tensor: Tensor,
    pub storage: Storage,
}

A tensor using the tch backend.

Fields

tensor: Tensor

Handle to the tensor. Call methods on this field.

storage: Storage

The tensor’s storage

Implementations

impl TchTensor

pub fn new(tensor: Tensor) -> Self

Create a new tensor.

Note that if the tensor was created from an operation that may reuse the same tensor storage as the parent, you should use from_existing instead.

pub fn from_existing(tensor: Tensor, storage_parent: Storage) -> Self

Create a tensor that was created from an operation executed on a parent tensor.

If the child tensor shared the same storage as its parent, it will be cloned, effectively tracking how much tensors point to the same memory space.

pub fn partial(tensor: Tensor, storage_parent: Storage) -> Self

Create a tensor that uses a part of its parent tensor such as slice and narrow.

impl TchTensor

pub fn can_mut(&self) -> bool

Checks if the tensor can be mutated in-place.

Returns true if the tensor’s stride does not contain zero (no broadcasting) and the storage can be mutated.

pub fn mut_ops<F: Fn(&mut Tensor) -> Tensor>( &mut self, func: F, ) -> Option<TchTensor>

Executes an operation on a tensor if the data can be reused.

pub fn unary_ops<FOwn, FRef>(self, fown: FOwn, fref: FRef) -> TchTensor

where FOwn: Fn(Tensor) -> Tensor, FRef: Fn(&Tensor) -> Tensor,

Executes a unary operation, reusing the tensor data if possible.

pub fn binary_ops_tensor<FLMut, FRMut, FRef>( lhs: Self, rhs: Self, flmut: FLMut, frmut: FRMut, fref: FRef, ) -> TchTensor

where FLMut: Fn(&mut Tensor, &Tensor) -> Tensor, FRMut: Fn(&Tensor, &mut Tensor) -> Tensor, FRef: Fn(&Tensor, &Tensor) -> Tensor,

Executes a binary operation, reusing the tensor data if possible.

impl TchTensor

pub fn from_data<E: TchElement>(data: TensorData, device: Device) -> Self

Creates a new tensor from a shape and a device.

Arguments

• data - The tensor’s data.
• device - The device on which the tensor will be allocated.

Returns

A new tensor.

impl TchTensor

pub fn empty<E: Element>(shape: Shape, device: LibTorchDevice) -> Self

Creates an empty tensor from a shape and a device.

Arguments

• shape - The shape of the tensor.
• device - The device to create the tensor on.

Returns

A new empty tensor.

Trait Implementations

impl Clone for TchTensor

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for TchTensor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for TchTensor

fn eq(&self, other: &TchTensor) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl TensorMetadata for TchTensor

fn dtype(&self) -> DType

The dtype of the tensor.

fn shape(&self) -> Shape

The shape of the tensor.

impl Send for TchTensor

impl StructuralPartialEq for TchTensor

impl Sync for TchTensor