TensorPart.h

Go to the documentation of this file.

#ifndef TENSORPART_H
#define TENSORPART_H
#pragma once
 
#include "txeo/Tensor.h"
 
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);
 
  private:
    TensorPart() = default;
};
 
class TensorPartError : public std::runtime_error {
  public:
    using std::runtime_error::runtime_error;
};
 
} // namespace txeo
 
#endif