Enum Op

pub enum Op {
    Binary(Tensor, Tensor, BinaryOp),
    Unary(Tensor, UnaryOp),
    Cmp(Tensor, CmpOp),
    Reduce(Tensor, ReduceOp, Vec<usize>),
    Matmul(Tensor, Tensor),
    Gather(Tensor, Tensor, usize),
    ScatterAdd(Tensor, Tensor, Tensor, usize),
    IndexSelect(Tensor, Tensor, usize),
    IndexAdd(Tensor, Tensor, Tensor, usize),
    WhereCond(Tensor, Tensor, Tensor),
    Conv1D {
        arg: Tensor,
        kernel: Tensor,
        padding: usize,
        stride: usize,
        dilation: usize,
    },
    ConvTranspose1D {
        arg: Tensor,
        kernel: Tensor,
        padding: usize,
        output_padding: usize,
        stride: usize,
        dilation: usize,
    },
    Conv2D {
        arg: Tensor,
        kernel: Tensor,
        padding: usize,
        stride: usize,
        dilation: usize,
    },
    ConvTranspose2D {
        arg: Tensor,
        kernel: Tensor,
        padding: usize,
        output_padding: usize,
        stride: usize,
        dilation: usize,
    },
    AvgPool2D {
        arg: Tensor,
        kernel_size: (usize, usize),
        stride: (usize, usize),
    },
    MaxPool2D {
        arg: Tensor,
        kernel_size: (usize, usize),
        stride: (usize, usize),
    },
    UpsampleNearest1D {
        arg: Tensor,
        target_size: usize,
    },
    UpsampleNearest2D {
        arg: Tensor,
        target_h: usize,
        target_w: usize,
    },
    Cat(Vec<Tensor>, usize),
    Affine {
        arg: Tensor,
        mul: f64,
        add: f64,
    },
    ToDType(Tensor),
    Copy(Tensor),
    Broadcast(Tensor),
    Narrow(Tensor, usize, usize, usize),
    SliceScatter0(Tensor, Tensor, usize),
    Reshape(Tensor),
    ToDevice(Tensor),
    Transpose(Tensor, usize, usize),
    Permute(Tensor, Vec<usize>),
    Elu(Tensor, f64),
    Powf(Tensor, f64),
    CustomOp1(Tensor, Arc<Box<dyn CustomOp1 + Send + Sync>>),
    CustomOp2(Tensor, Tensor, Arc<Box<dyn CustomOp2 + Send + Sync>>),
    CustomOp3(Tensor, Tensor, Tensor, Arc<Box<dyn CustomOp3 + Send + Sync>>),
}

Variants

Binary(Tensor, Tensor, BinaryOp)

Unary(Tensor, UnaryOp)

Cmp(Tensor, CmpOp)

Reduce(Tensor, ReduceOp, Vec<usize>)

Matmul(Tensor, Tensor)

Gather(Tensor, Tensor, usize)

ScatterAdd(Tensor, Tensor, Tensor, usize)

IndexSelect(Tensor, Tensor, usize)

IndexAdd(Tensor, Tensor, Tensor, usize)

WhereCond(Tensor, Tensor, Tensor)

Conv1D

Fields

arg: Tensor

kernel: Tensor

padding: usize

stride: usize

dilation: usize

ConvTranspose1D

Fields

arg: Tensor

kernel: Tensor

padding: usize

output_padding: usize

stride: usize

dilation: usize

Conv2D

Fields

arg: Tensor

kernel: Tensor

padding: usize

stride: usize

dilation: usize

ConvTranspose2D

Fields

arg: Tensor

kernel: Tensor

padding: usize

output_padding: usize

stride: usize

dilation: usize

AvgPool2D

Fields

arg: Tensor

kernel_size: (usize, usize)

stride: (usize, usize)

MaxPool2D

Fields

arg: Tensor

kernel_size: (usize, usize)

stride: (usize, usize)

UpsampleNearest1D

Fields

arg: Tensor

target_size: usize

UpsampleNearest2D

Fields

arg: Tensor

target_h: usize

target_w: usize

Cat(Vec<Tensor>, usize)

Affine

Fields

arg: Tensor

mul: f64

add: f64

ToDType(Tensor)

Copy(Tensor)

Broadcast(Tensor)

Narrow(Tensor, usize, usize, usize)

SliceScatter0(Tensor, Tensor, usize)

Reshape(Tensor)

ToDevice(Tensor)

Transpose(Tensor, usize, usize)

Permute(Tensor, Vec<usize>)

Elu(Tensor, f64)

Powf(Tensor, f64)

CustomOp1(Tensor, Arc<Box<dyn CustomOp1 + Send + Sync>>)

CustomOp2(Tensor, Tensor, Arc<Box<dyn CustomOp2 + Send + Sync>>)

CustomOp3(Tensor, Tensor, Tensor, Arc<Box<dyn CustomOp3 + Send + Sync>>)

Trait Implementations

impl Clone for Op

fn clone(&self) -> Op

Returns a copy of the value.

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

Performs copy-assignment from source.