page_ops.hpp

Go to the documentation of this file.

// toolboxes/linalg/include/nk/linalg/page_ops.hpp
//
// Page-wise linalg ops on 3-D arrays. Wrappers that iterate over
// pages and dispatch to the 2-D variant. The general page operators
// (pagetranspose, pagectranspose, pagemtimes) stay in toolboxes/builtin —
// they're basic tensor ops, not strictly linalg.
 
#pragma once
 
#include <memory_resource>
#include <tuple>
#include <nk/value/value.hpp>
 
namespace nk::linalg {

 

Value pageinv(const Value &A, std::pmr::memory_resource *mr = nullptr);

Value pageeig_values(const Value &A, std::pmr::memory_resource *mr = nullptr);

std::tuple<Value, Value>
pageeig_VD(const Value &A, std::pmr::memory_resource *mr = nullptr);

Value pagesvd_values(const Value &A, std::pmr::memory_resource *mr = nullptr);

std::tuple<Value, Value, Value>
pagesvd_decompose(const Value &A, std::pmr::memory_resource *mr = nullptr);

Value pagepinv(const Value &A, double tol = -1.0,
               std::pmr::memory_resource *mr = nullptr);

Value pagenorm(const Value &A, double p, std::pmr::memory_resource *mr = nullptr);

Value pagemldivide(const Value &A, const Value &B, std::pmr::memory_resource *mr = nullptr);

Value pagemrdivide(const Value &A, const Value &B, std::pmr::memory_resource *mr = nullptr);

Value pagelsqminnorm(const Value &A, const Value &B, bool have_tol, double tol_user,
                     std::pmr::memory_resource *mr = nullptr);
 

} // namespace nk::linalg