torchUTILITIES.C File Reference

#include "torchUTILITIES.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)