Class SpMMOp

Namespace

AiDotNet.JitCompiler.IR.Operations

Assembly

AiDotNet.dll

Represents sparse matrix-matrix multiplication in the IR.

public class SpMMOp : IROp

Inheritance

object

IROp

SpMMOp

Inherited Members

IROp.OutputIds

IROp.OutputId

IROp.InputIds

IROp.OutputType

IROp.OutputShape

IROp.OutputShapes

IROp.OpType

IROp.ToString()

object.Equals(object)

object.Equals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

object.ReferenceEquals(object, object)

Remarks

SpMM computes C = A * B where A is a sparse matrix and B is a dense matrix. This is essential for batched graph neural network operations.

For Beginners: Like SpMV but the second operand is a matrix instead of a vector. This allows processing multiple feature vectors at once, which is common in neural network training with batches.

Properties

DenseColumns

Gets or sets the number of columns in the dense matrix.

public int DenseColumns { get; set; }

Property Value

int

NonZeroCount

Gets or sets the number of non-zero elements.

public int NonZeroCount { get; set; }

Property Value

int

SparseColumns

Gets or sets the number of columns in the sparse matrix.

public int SparseColumns { get; set; }

Property Value

int

SparseRows

Gets or sets the number of rows in the sparse matrix.

public int SparseRows { get; set; }

Property Value

int

Methods

Validate()

Validates that this operation is correctly formed.

public override bool Validate()

Returns

bool

True if valid, false otherwise.