TensorFunc.h

Go to the documentation of this file.

#ifndef TENSORFUNC_H
#define TENSORFUNC_H
#pragma once
 
#include "txeo/Matrix.h"
#include "txeo/Tensor.h"
 
#include <cstddef>
 
namespace txeo {
 
template <typename T>
class TensorFunc {
  public:
    TensorFunc(const TensorFunc &) = delete;
    TensorFunc(TensorFunc &&) = delete;
    TensorFunc &operator=(const TensorFunc &) = default;
    TensorFunc &operator=(TensorFunc &&) = delete;
    ~TensorFunc() = default;
 
    static txeo::Tensor<T> power_elem(const txeo::Tensor<T> &tensor, const T &exponent);
 
    static txeo::Tensor<T> power_elem_by(txeo::Tensor<T> &tensor, const T &exponent);
 
    static txeo::Tensor<T> square(const txeo::Tensor<T> &tensor);
 
    static txeo::Tensor<T> &square_by(txeo::Tensor<T> &tensor);
 
    static txeo::Tensor<T> sqrt(const txeo::Tensor<T> &tensor);
 
    static txeo::Tensor<T> &sqrt_by(txeo::Tensor<T> &tensor);
 
    static txeo::Tensor<T> abs(const txeo::Tensor<T> &tensor);
 
    static txeo::Tensor<T> &abs_by(txeo::Tensor<T> &tensor);
 
    static txeo::Tensor<T> permute(const txeo::Tensor<T> &tensor, const std::vector<size_t> &axes);
 
    static txeo::Tensor<T> &permute_by(txeo::Tensor<T> &tensor, const std::vector<size_t> &axes);
 
    static txeo::Matrix<T> transpose(const txeo::Matrix<T> &matrix);
 
    static txeo::Matrix<T> &transpose_by(txeo::Matrix<T> &matrix);
 
  private:
    TensorFunc() = default;
};
 
class TensorFuncError : public std::runtime_error {
  public:
    using std::runtime_error::runtime_error;
};
 
} // namespace txeo
 
#endif