/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor.hpp Source File#

5 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/tensor.hpp Source File

Go to the documentation of this file.

  
 #pragma once
  
 #include <memory>
  
 #include "tensor_data.hpp"
 #include "tensor_requirements.hpp"
 #include "tensor_storage.hpp"
  
 namespace roccv {
  
 class ImageFormat;
 struct Size2D;
 class TensorShape;
 class TensorLayout;
  
 class Tensor {
    public:
     using Requirements = TensorRequirements;
  
     explicit Tensor(const TensorRequirements &reqs);
  
     explicit Tensor(const TensorRequirements &reqs, std::shared_ptr<TensorStorage> data);
  
     explicit Tensor(const TensorShape &shape, DataType dtype, const eDeviceType device = eDeviceType::GPU);
  
     explicit Tensor(int num_images, Size2D image_size, ImageFormat fmt, eDeviceType device = eDeviceType::GPU);
  
     Tensor(const Tensor &other) = delete;
     Tensor(Tensor &&other);
  
     int rank() const;
  
     const eDeviceType device() const;
  
     const TensorShape &shape() const;
  
     const int64_t shape(int d) const &;
  
     const DataType &dtype() const;
  
     const TensorLayout &layout() const;
  
     TensorData exportData() const;
  
     template <typename DerivedTensorData>
     DerivedTensorData exportData() const {
         TensorData data = exportData();
         std::optional<DerivedTensorData> derived_tensor = data.cast<DerivedTensorData>();
         if (!derived_tensor.has_value()) {
             throw std::bad_cast();
         }
  
         return derived_tensor.value();
     }
  
     Tensor reshape(const TensorShape &new_shape) const;
  
     Tensor reshape(const TensorShape &new_shape, const DataType &new_dtype) const;
  
     Tensor &operator=(const Tensor &other);
  
     static Requirements CalcRequirements(const TensorShape &shape, DataType dtype,
                                          const eDeviceType device = eDeviceType::GPU);
  
     static Requirements CalcRequirements(int num_images, Size2D image_size, ImageFormat fmt,
                                          eDeviceType device = eDeviceType::GPU);
  
    private:
     TensorRequirements m_requirements;      // Tensor metadata
     std::shared_ptr<TensorStorage> m_data;  // Stores raw tensor data
 };
  
 extern Tensor TensorWrapData(const TensorData &tensor_data);
  
 }  // namespace roccv