Operators

Linear operator abstractions for GPJax.

LinearOperator

LinearOperator()

Bases: ABC

Abstract base class for linear operators.

shape abstractmethod property

shape

Return the shape of the operator.

dtype abstractmethod property

dtype

Return the data type of the operator.

T property

T

Return the transpose of the operator.

to_dense abstractmethod

to_dense()

Convert the operator to a dense JAX array.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).

Dense

Dense(array)

Bases: LinearOperator

Dense linear operator wrapping a JAX array.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).

Diagonal

Diagonal(diagonal)

Bases: LinearOperator

Diagonal linear operator.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).

Identity

Identity(shape, dtype=jnp.float64)

Bases: LinearOperator

Identity linear operator.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).

Triangular

Triangular(array, lower=True)

Bases: LinearOperator

Triangular linear operator.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).

BlockDiag

BlockDiag(operators, multiplicities=None)

Bases: LinearOperator

Block diagonal linear operator.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).

Kronecker

Kronecker(operators)

Bases: LinearOperator

Kronecker product linear operator.

__add__

__add__(other)

Addition with another array or operator.

__mul__

__mul__(other)

Scalar multiplication (self * scalar).