einsum.h File Reference

#include <string>
#include <tuple>
#include <vector>
#include "mlx/array.h"
#include "mlx/utils.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Functions
std::pair< std::vector< std::vector< int > >, std::string >	mlx::core::einsum_path (const std::string &subscripts, const std::vector< array > &operands)
array	mlx::core::einsum (const std::string &subscripts, const std::vector< array > &operands, StreamOrDevice s={})