TensorPart.h

Go to the documentation of this file.

#ifndef TENSORPART_H
#define TENSORPART_H
#pragma once
 
#include "txeo/Matrix.h"
#include "txeo/Tensor.h"
 
#include <cstddef>
#include <stdexcept>
#include <vector>
 
namespace txeo {
 
template <typename T>
class TensorPart {
  public:
    TensorPart(const TensorPart &) = delete;
    TensorPart(TensorPart &&) = delete;
    TensorPart &operator=(const TensorPart &) = delete;
    TensorPart &operator=(TensorPart &&) = delete;
    ~TensorPart();
 
    static std::vector<txeo::Tensor<T>> unstack(const txeo::Tensor<T> &tensor, size_t axis);
 
    static txeo::Tensor<T> slice(const txeo::Tensor<T> &tensor, size_t first_axis_begin,
                                 size_t first_axis_end);
 
    static txeo::Tensor<T> increase_dimension(const txeo::Tensor<T> &tensor, size_t axis, T value);
 
    static txeo::Tensor<T> &increase_dimension_by(txeo::Tensor<T> &tensor, size_t axis, T value);
 
    static txeo::Matrix<T> sub_matrix_cols(const txeo::Matrix<T> &matrix,
                                           const std::vector<size_t> &cols);
 
    static txeo::Matrix<T> sub_matrix_cols_exclude(const txeo::Matrix<T> &matrix,
                                                   const std::vector<size_t> &cols);
 
    static txeo::Matrix<T> sub_matrix_rows(const txeo::Matrix<T> &matrix,
                                           const std::vector<size_t> &rows);
 
  private:
    TensorPart() = default;
};
 
class TensorPartError : public std::runtime_error {
  public:
    using std::runtime_error::runtime_error;
};
 
} // namespace txeo
 
#endif