tensor_map.h

Go to the documentation of this file.

#ifndef ATEN_CORE_TENSOR_MAP_H_
#define ATEN_CORE_TENSOR_MAP_H_
 
#include <complex>
 
#include <ATen/core/tensor.h>
 
namespace container {
 
 class TensorMap : public Tensor {
 public:
 
    TensorMap(void *data, DataType data_type, DeviceType device, const TensorShape &shape);
 
    TensorMap(void *data, const Tensor& other, const TensorShape& shape);
 
    TensorMap(void *data, const Tensor& other);
 
};
 
} // namespace container
 
#endif // ATEN_CORE_TENSOR_MAP_H_