Enum StdTensorOp 

pub enum StdTensorOp {
    Add,
    Mul,
    Neg,
    Conj,
    DotGeneral {
        config: DotGeneralConfig,
    },
    Transpose {
        perm: Vec<usize>,
    },
    Reshape {
        to_shape: Vec<DimExpr>,
    },
    BroadcastInDim {
        shape: Vec<DimExpr>,
        dims: Vec<usize>,
    },
    Convert {
        from: DType,
        to: DType,
    },
    Constant {
        dtype: DType,
        bytes: Vec<u8>,
    },
    ReduceSum {
        axes: Vec<usize>,
    },
    Div,
    Abs,
    Sign,
    Maximum,
    Minimum,
    Compare(CompareDir),
    Select,
    Clamp,
    Exp,
    Log,
    Sin,
    Cos,
    Tanh,
    Sqrt,
    Rsqrt,
    Pow,
    Expm1,
    Log1p,
    ExtractDiag {
        axis_a: usize,
        axis_b: usize,
    },
    EmbedDiag {
        axis_a: usize,
        axis_b: usize,
    },
    Tril {
        k: i64,
    },
    Triu {
        k: i64,
    },
    Gather(GatherConfig),
    GatherDynamicSliceSizes {
        offset_dims: Vec<usize>,
        collapsed_slice_dims: Vec<usize>,
        start_index_map: Vec<usize>,
        index_vector_dim: usize,
        slice_sizes: Vec<DimExpr>,
    },
    Scatter(ScatterConfig),
    Slice(SliceConfig),
    DynamicSlice {
        slice_sizes: Vec<usize>,
    },
    DynamicUpdateSlice,
    Pad(PadConfig),
    Concatenate {
        axis: usize,
        input_count: usize,
    },
    Reverse {
        axes: Vec<usize>,
    },
    ShapeOf {
        axis: usize,
    },
    DynamicTruncate {
        axis: usize,
    },
    PadToMatch {
        axis: usize,
    },
    ReduceProd {
        axes: Vec<usize>,
    },
    ReduceMax {
        axes: Vec<usize>,
    },
    ReduceMin {
        axes: Vec<usize>,
    },
    Extension(Arc<dyn ExtensionOp>),
}

Variants

Add

Mul

Neg

Conj

DotGeneral

Fields

config: DotGeneralConfig

Transpose

Fields

perm: Vec<usize>

Reshape

Fields

to_shape: Vec<DimExpr>

BroadcastInDim

Fields

shape: Vec<DimExpr>

dims: Vec<usize>

Convert

Fields

from: DType

to: DType

Constant

Fields

dtype: DType

bytes: Vec<u8>

ReduceSum

Fields

axes: Vec<usize>

Div

Abs

Sign

Maximum

Minimum

Compare(CompareDir)

Select

Clamp

Exp

Log

Sin

Cos

Tanh

Sqrt

Rsqrt

Pow

Expm1

Log1p

ExtractDiag

Fields

axis_a: usize

axis_b: usize

EmbedDiag

Fields

axis_a: usize

axis_b: usize

Tril

Fields

k: i64

Triu

Fields

k: i64

Gather(GatherConfig)

GatherDynamicSliceSizes

Fields

offset_dims: Vec<usize>

collapsed_slice_dims: Vec<usize>

start_index_map: Vec<usize>

index_vector_dim: usize

slice_sizes: Vec<DimExpr>

Scatter(ScatterConfig)

Slice(SliceConfig)

DynamicSlice

Fields

slice_sizes: Vec<usize>

DynamicUpdateSlice

Pad(PadConfig)

Concatenate

Fields

axis: usize

input_count: usize

Reverse

Fields

axes: Vec<usize>

ShapeOf

Fields

axis: usize

DynamicTruncate

Fields

axis: usize

PadToMatch

Fields

axis: usize

ReduceProd

Fields

axes: Vec<usize>

ReduceMax

Fields

axes: Vec<usize>

ReduceMin

Fields

axes: Vec<usize>

Extension(Arc<dyn ExtensionOp>)

Out-of-tree extension carrier.

See crate::ext_op and docs/spec/extension-op.md. Identity, hashing, equality, arity, shape inference, and AD rules are delegated to the inner ExtensionOp trait object.

Implementations

impl StdTensorOp

pub fn primitive_kind(&self) -> Option<PrimitiveOpKind>

Return the core primitive catalog kind for this graph operation.

Extension operations do not claim a core primitive kind; they are dispatched through their extension family id instead.

Examples

use tenferro_core_ops::PrimitiveOpKind;
use tenferro_ops::std_tensor_op::StdTensorOp;

assert_eq!(StdTensorOp::Add.primitive_kind(), Some(PrimitiveOpKind::Add));

impl StdTensorOp

pub fn constant<T: ConstantScalar>(value: T) -> Self

Create a scalar constant op from any supported tensor scalar.

Examples

use num_complex::Complex64;
use tenferro_ops::std_tensor_op::StdTensorOp;
use tenferro_tensor::DType;

let real = StdTensorOp::constant(1.5_f64);
let complex = StdTensorOp::constant(Complex64::new(1.0, -2.0));

assert!(matches!(real, StdTensorOp::Constant { dtype: DType::F64, .. }));
assert!(matches!(complex, StdTensorOp::Constant { dtype: DType::C64, .. }));

Trait Implementations

impl Clone for StdTensorOp

fn clone(&self) -> StdTensorOp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StdTensorOp

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

Formats the value using the given formatter.

impl GraphOperation for StdTensorOp

type Operand = Tensor

type Context = ()

type InputKey = TensorInputKey

fn input_count(&self) -> usize

Returns the number of inputs consumed by this operation.

fn output_count(&self) -> usize

Returns the number of outputs produced by this operation.

impl Hash for StdTensorOp

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

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 PartialEq for StdTensorOp

fn eq(&self, other: &Self) -> 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 Primitive for StdTensorOp

Available on crate feature autodiff only.

type ADContext = ShapeGuardContext

Runtime AD context threaded through linearization and transposition.

fn add() -> Self

Returns the addition operation used for cotangent accumulation in [crate::linear_transpose]. When multiple cotangents flow to the same ValueKey, linear_transpose emits Op::add() nodes to sum them.

fn jvp_rule( &self, builder: &mut impl PrimitiveBuilder<Self>, primal_in: &[ValueKey<Self>], primal_out: &[ValueKey<Self>], tangent_in: &[Option<LocalValueId>], ctx: &mut Self::ADContext, ) -> ADRuleResult<Vec<Option<LocalValueId>>>

Emit the JVP rule for this primitive.

fn transpose_rule( &self, builder: &mut impl PrimitiveBuilder<Self>, cotangent_out: &[Option<LocalValueId>], inputs: &[PrimitiveValue<Self>], mode: &OperationRole, ctx: &mut Self::ADContext, ) -> ADRuleResult<Vec<Option<LocalValueId>>>

Emit the transpose rule for this linear primitive.

impl Eq for StdTensorOp