torchUTILITIES.H File Reference

Header file of the torchUTILITIES file. More...

#include <iostream>
#include <torch/script.h>
#include <torch/torch.h>
#include "ITHACAassert.H"
#include "cnpy.H"

Go to the source code of this file.

Namespaces
namespace	ITHACAtorch

Functions
torch::Tensor	ITHACAtorch::removeConstValues (torch::Tensor input, std::vector< int > &indices, std::vector< double > &constValues)
bool	ITHACAtorch::isConst (torch::Tensor &tTensor)
void	ITHACAtorch::save (const torch::Tensor &torchTensor, const std::string fname)
torch::Tensor	ITHACAtorch::load (const std::string fname)

Detailed Description

Header file of the torchUTILITIES file.

It contains functions to manipulate torch tensors

Definition in file torchUTILITIES.H.