Struct TracedTensor 

pub struct TracedTensor { /* private fields */ }

A traced tensor — records ops instead of executing them.

Implementations

impl TracedTensor

pub fn matmul(self, rhs: TracedTensor) -> TracedTensor

pub fn gelu(self) -> TracedTensor

pub fn silu(self) -> TracedTensor

pub fn relu(self) -> TracedTensor

pub fn layer_norm( self, gamma: TracedTensor, beta: TracedTensor, eps: f32, ) -> TracedTensor

pub fn softmax(self, axis: i32) -> TracedTensor

pub fn reshape(self, new_shape: &[i64]) -> TracedTensor

pub fn transpose(self, perm: &[usize]) -> TracedTensor

pub fn narrow(self, axis: usize, start: usize, len: usize) -> TracedTensor

pub fn rank(&self) -> usize

Number of dimensions. PyTorch’s .dim().

pub fn shape(&self) -> Shape

Output shape — useful for derived computations / asserts.

pub fn t(&self) -> TracedTensor

2-D transpose shorthand. t.t() swaps the last two axes, matching PyTorch’s .t() for matrices.

pub fn permute(&self, perm: &[usize]) -> TracedTensor

Permute dimensions — alias of Self::transpose under PyTorch’s name.

pub fn unsqueeze(&self, axis: usize) -> TracedTensor

Insert a length-1 dim at axis. Bumps every existing dim at >= axis by one position.

pub fn squeeze(&self, axis: usize) -> TracedTensor

Drop a length-1 dim at axis. Errors at compile-time-of- graph if the dim isn’t 1.

pub fn mm(&self, rhs: &TracedTensor) -> TracedTensor

Reference-friendly matmul. a.mm(&b) doesn’t move either.

Trait Implementations

impl Add<&TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the + operator.

fn add(self, rhs: &TracedTensor) -> TracedTensor

Performs the + operation.

impl Add<&TracedTensor> for TracedTensor

type Output = TracedTensor

The resulting type after applying the + operator.

fn add(self, rhs: &TracedTensor) -> TracedTensor

Performs the + operation.

impl Add<TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the + operator.

fn add(self, rhs: TracedTensor) -> TracedTensor

Performs the + operation.

impl Add for TracedTensor

type Output = TracedTensor

The resulting type after applying the + operator.

fn add(self, rhs: TracedTensor) -> TracedTensor

Performs the + operation.

impl Clone for TracedTensor

fn clone(&self) -> TracedTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Div<&TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the / operator.

fn div(self, rhs: &TracedTensor) -> TracedTensor

Performs the / operation.

impl Div<&TracedTensor> for TracedTensor

type Output = TracedTensor

The resulting type after applying the / operator.

fn div(self, rhs: &TracedTensor) -> TracedTensor

Performs the / operation.

impl Div<TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the / operator.

fn div(self, rhs: TracedTensor) -> TracedTensor

Performs the / operation.

impl Div for TracedTensor

type Output = TracedTensor

The resulting type after applying the / operator.

fn div(self, rhs: TracedTensor) -> TracedTensor

Performs the / operation.

impl Mul<&TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the * operator.

fn mul(self, rhs: &TracedTensor) -> TracedTensor

Performs the * operation.

impl Mul<&TracedTensor> for TracedTensor

type Output = TracedTensor

The resulting type after applying the * operator.

fn mul(self, rhs: &TracedTensor) -> TracedTensor

Performs the * operation.

impl Mul<TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the * operator.

fn mul(self, rhs: TracedTensor) -> TracedTensor

Performs the * operation.

impl Mul for TracedTensor

type Output = TracedTensor

The resulting type after applying the * operator.

fn mul(self, rhs: TracedTensor) -> TracedTensor

Performs the * operation.

impl Neg for &TracedTensor

type Output = TracedTensor

The resulting type after applying the - operator.

fn neg(self) -> TracedTensor

Performs the unary - operation.

impl Neg for TracedTensor

type Output = TracedTensor

The resulting type after applying the - operator.

fn neg(self) -> TracedTensor

Performs the unary - operation.

impl Sub<&TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the - operator.

fn sub(self, rhs: &TracedTensor) -> TracedTensor

Performs the - operation.

impl Sub<&TracedTensor> for TracedTensor

type Output = TracedTensor

The resulting type after applying the - operator.

fn sub(self, rhs: &TracedTensor) -> TracedTensor

Performs the - operation.

impl Sub<TracedTensor> for &TracedTensor

type Output = TracedTensor

The resulting type after applying the - operator.

fn sub(self, rhs: TracedTensor) -> TracedTensor

Performs the - operation.

impl Sub for TracedTensor

type Output = TracedTensor

The resulting type after applying the - operator.

fn sub(self, rhs: TracedTensor) -> TracedTensor

Performs the - operation.