linalg.h File Reference

#include <optional>
#include "mlx/array.h"
#include "mlx/device.h"
#include "mlx/ops.h"
#include "mlx/stream.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::linalg

Functions
array	mlx::core::linalg::norm (const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})
	Compute vector or matrix norms.
array	mlx::core::linalg::norm (const array &a, const double ord, int axis, bool keepdims=false, StreamOrDevice s={})
array	mlx::core::linalg::norm (const array &a, const std::string &ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})
array	mlx::core::linalg::norm (const array &a, const std::string &ord, int axis, bool keepdims=false, StreamOrDevice s={})
array	mlx::core::linalg::norm (const array &a, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})
array	mlx::core::linalg::norm (const array &a, int axis, bool keepdims=false, StreamOrDevice s={})
std::pair< array, array >	mlx::core::linalg::qr (const array &a, StreamOrDevice s={})
std::vector< array >	mlx::core::linalg::svd (const array &a, StreamOrDevice s={})
array	mlx::core::linalg::inv (const array &a, StreamOrDevice s={})
array	mlx::core::linalg::tri_inv (const array &a, bool upper=false, StreamOrDevice s={})
array	mlx::core::linalg::cholesky (const array &a, bool upper=false, StreamOrDevice s={})
array	mlx::core::linalg::pinv (const array &a, StreamOrDevice s={})
array	mlx::core::linalg::cholesky_inv (const array &a, bool upper=false, StreamOrDevice s={})
array	mlx::core::linalg::cross (const array &a, const array &b, int axis=-1, StreamOrDevice s={})
	Compute the cross product of two arrays along the given axis.