neml2::math::linalg Namespace Reference

Functions
BatchTensor	vector_norm (const BatchTensor &v)
	Vector norm of a vector. Falls back to math::abs is v is a Scalar.
BatchTensor	inv (const BatchTensor &m)
	Inverse of a square matrix.
BatchTensor	solve (const BatchTensor &A, const BatchTensor &B)
	Solve the linear system A X = B.
std::tuple< BatchTensor, BatchTensor >	lu_factor (const BatchTensor &A, bool pivot)
BatchTensor	lu_solve (const BatchTensor &LU, const BatchTensor &pivots, const BatchTensor &B, bool left, bool adjoint)

Function Documentation

inv()

BatchTensor inv

(

const BatchTensor &

m

)

Inverse of a square matrix.

lu_factor()

std::tuple< BatchTensor, BatchTensor > lu_factor	(	const BatchTensor &	A,
		bool	pivot )

lu_solve()

BatchTensor lu_solve	(	const BatchTensor &	LU,
		const BatchTensor &	pivots,
		const BatchTensor &	B,
		bool	left,
		bool	adjoint )

solve()

BatchTensor solve	(	const BatchTensor &	A,
		const BatchTensor &	B )

Solve the linear system A X = B.

vector_norm()

BatchTensor vector_norm

(

const BatchTensor &

v

)

Vector norm of a vector. Falls back to math::abs is v is a Scalar.