Header file of the torch2Eigen namespace. More...
#include <iostream>
#include <torch/script.h>
#include <torch/torch.h>
#include <Eigen/Eigen>
#include "ITHACAassert.H"
Go to the source code of this file.
Namespaces | |
namespace | ITHACAtorch |
namespace | ITHACAtorch::torch2Eigen |
Functions | |
template<class type > | |
torch::Tensor | ITHACAtorch::torch2Eigen::eigenMatrix2torchTensor (Eigen::Matrix< type, Eigen::Dynamic, Eigen::Dynamic > eigenMatrix) |
Convert an eigen Matrix to a torch tensor. | |
template<class type > | |
Eigen::Matrix< type, Eigen::Dynamic, Eigen::Dynamic > | ITHACAtorch::torch2Eigen::torchTensor2eigenMatrix (torch::Tensor &torchTensor) |
Convert a torch tensor to an eigen Matrix. | |
Header file of the torch2Eigen namespace.
It contains functions to
Definition in file torch2Eigen.H.