Enum burn::tensor::repr::NumericOperationDescription

pub enum NumericOperationDescription<E> {
    Add(BinaryOperationDescription),
    AddScalar(ScalarOperationDescription<E>),
    Sub(BinaryOperationDescription),
    SubScalar(ScalarOperationDescription<E>),
    Div(BinaryOperationDescription),
    DivScalar(ScalarOperationDescription<E>),
    RemScalar(ScalarOperationDescription<E>),
    Mul(BinaryOperationDescription),
    MulScalar(ScalarOperationDescription<E>),
    Abs(UnaryOperationDescription),
    Ones(TensorDescription),
    Zeros(TensorDescription),
    Full((TensorDescription, E)),
    Gather(GatherOperationDescription),
    Scatter(ScatterOperationDescription),
    Select(SelectOperationDescription),
    SelectAssign(SelectAssignOperationDescription),
    MaskWhere(MaskWhereOperationDescription),
    MaskFill(MaskFillOperationDescription<E>),
    MeanDim(ScalarOperationDescription<usize>),
    Mean(UnaryOperationDescription),
    Sum(UnaryOperationDescription),
    SumDim(ScalarOperationDescription<usize>),
    Prod(UnaryOperationDescription),
    ProdDim(ScalarOperationDescription<usize>),
    EqualElem(ScalarOperationDescription<E>),
    Greater(BinaryOperationDescription),
    GreaterElem(ScalarOperationDescription<E>),
    GreaterEqual(BinaryOperationDescription),
    GreaterEqualElem(ScalarOperationDescription<E>),
    Lower(BinaryOperationDescription),
    LowerElem(ScalarOperationDescription<E>),
    LowerEqual(BinaryOperationDescription),
    LowerEqualElem(ScalarOperationDescription<E>),
    ArgMax(ScalarOperationDescription<usize>),
    ArgMin(ScalarOperationDescription<usize>),
    Max(UnaryOperationDescription),
    MaxDimWithIndices(ReduceDimWithIndicesDescription),
    MinDimWithIndices(ReduceDimWithIndicesDescription),
    Min(UnaryOperationDescription),
    MaxDim(ScalarOperationDescription<usize>),
    MinDim(ScalarOperationDescription<usize>),
    Clamp(ClampOperationDescription<E>),
    IntRandom(RandomOperationDescription),
    Powf(BinaryOperationDescription),
}

Numeric operations on int and float tensors.

Variants

Add(BinaryOperationDescription)

Operation corresponding to:

Float => add. Int => add.

AddScalar(ScalarOperationDescription<E>)

Operation corresponding to:

Float => add scalar. Int => add scalar.

Sub(BinaryOperationDescription)

Operation corresponding to:

Float => sub. Int => sub.

SubScalar(ScalarOperationDescription<E>)

Operation corresponding to:

Float => sub scalar. Int => sub scalar.

Div(BinaryOperationDescription)

Operation corresponding to:

Float => div. Int => div.

DivScalar(ScalarOperationDescription<E>)

Operation corresponding to:

Float => div scalar. Int => div scalar.

RemScalar(ScalarOperationDescription<E>)

Operation corresponding to:

Float => div. Int => div.

Mul(BinaryOperationDescription)

Operation corresponding to:

Float => mul. Int => mul.

MulScalar(ScalarOperationDescription<E>)

Operation corresponding to:

Float => mul scalar. Int => mul scalar.

Abs(UnaryOperationDescription)

Operation corresponding to:

Float => abs. Int => abs.

Ones(TensorDescription)

Operation corresponding to:

Float => ones. Int => ones.

Zeros(TensorDescription)

Operation corresponding to:

Float => zeros. Int => zeros.

Full((TensorDescription, E))

Operation corresponding to:

Float => full. Int => full.

Gather(GatherOperationDescription)

Operation corresponding to:

Float => gather. Int => gather.

Scatter(ScatterOperationDescription)

Operation corresponding to:

Float => scatter. Int => scatter.

Select(SelectOperationDescription)

Operation corresponding to:

Float => select. Int => select.

SelectAssign(SelectAssignOperationDescription)

Operation corresponding to:

Float => select assign. Int => select assign.

MaskWhere(MaskWhereOperationDescription)

Operation corresponding to:

Float => mask where. Int => mask where.

MaskFill(MaskFillOperationDescription<E>)

Operation corresponding to:

Float => mask fill. Int => mask fill.

MeanDim(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => mean dim. Int => mean dim.

Mean(UnaryOperationDescription)

Operation corresponding to:

Float => mean. Int => mean.

Sum(UnaryOperationDescription)

Operation corresponding to:

Float => sum. Int => sum.

SumDim(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => sum dim. Int => sum dim.

Prod(UnaryOperationDescription)

Operation corresponding to:

Float => prod. Int => prod.

ProdDim(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => prod dim. Int => prod dim.

EqualElem(ScalarOperationDescription<E>)

Operation corresponding to:

Float => equal elem. Int => equal elem.

Greater(BinaryOperationDescription)

Operation corresponding to:

Float => greater. Int => greater.

GreaterElem(ScalarOperationDescription<E>)

Operation corresponding to:

Float => greater elem. Int => greater elem.

GreaterEqual(BinaryOperationDescription)

Operation corresponding to:

Float => greater equal. Int => greater elem.

GreaterEqualElem(ScalarOperationDescription<E>)

Operation corresponding to:

Float => greater equal elem. Int => greater equal elem.

Lower(BinaryOperationDescription)

Operation corresponding to:

Float => lower. Int => lower.

LowerElem(ScalarOperationDescription<E>)

Operation corresponding to:

Float => lower elem. Int => lower elem.

LowerEqual(BinaryOperationDescription)

Operation corresponding to:

Float => lower equal. Int => lower equal.

LowerEqualElem(ScalarOperationDescription<E>)

Operation corresponding to:

Float => lower equal elem. Int => lower equal elem.

ArgMax(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => argmax. Int => argmax.

ArgMin(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => argmin. Int => argmin.

Max(UnaryOperationDescription)

Operation corresponding to:

Float => max. Int => max.

MaxDimWithIndices(ReduceDimWithIndicesDescription)

Operation corresponding to:

Float => max dim with indices. Int => max dim with indices.

MinDimWithIndices(ReduceDimWithIndicesDescription)

Operation corresponding to:

Float => min dim with indices. Int => min dim with indices.

Min(UnaryOperationDescription)

Operation corresponding to:

Float => min. Int => min.

MaxDim(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => max dim. Int => max dim.

MinDim(ScalarOperationDescription<usize>)

Operation corresponding to:

Float => min dim. Int => min dim.

Clamp(ClampOperationDescription<E>)

Operation corresponding to:

Float => clamp. Int => clamp.

IntRandom(RandomOperationDescription)

Operation corresponding to:

Int => random.

Powf(BinaryOperationDescription)

Operation corresponding to:

Float => powf. Int => powf.

Trait Implementations

impl<E> Clone for NumericOperationDescription<E>

where E: Clone,

fn clone(&self) -> NumericOperationDescription<E>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<E> Debug for NumericOperationDescription<E>

where E: Debug,

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

Formats the value using the given formatter.

impl<'de, E> Deserialize<'de> for NumericOperationDescription<E>

where E: Deserialize<'de>,

fn deserialize<__D>( __deserializer: __D, ) -> Result<NumericOperationDescription<E>, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<E> Hash for NumericOperationDescription<E>

fn hash<H>(&self, state: &mut H)

where H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl<E> PartialEq for NumericOperationDescription<E>

where E: PartialEq,

fn eq(&self, other: &NumericOperationDescription<E>) -> bool

This method tests for self and other values to be equal, and is used by ==.

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

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

impl<E> Serialize for NumericOperationDescription<E>

where E: Serialize,

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl<E> StructuralPartialEq for NumericOperationDescription<E>