Loading...
Searching...
No Matches
ITHACAtorch Namespace Reference

Namespaces

namespace  torch2Eigen
 
namespace  torch2Foam
 

Classes

class  ConvLayer
 

Typedefs

typedef ConvLayer< scalar, fvPatchField, volMesh > volScalarLayer
 
typedef ConvLayer< vector, fvPatchField, volMesh > volVectorLayer
 
typedef ConvLayer< scalar, fvsPatchField, surfaceMesh > surfaceScalarLayer
 

Functions

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

Typedef Documentation

◆ surfaceScalarLayer

typedef ConvLayer<scalar, fvsPatchField, surfaceMesh> ITHACAtorch::surfaceScalarLayer

Definition at line 90 of file ConvLayer.H.

◆ volScalarLayer

typedef ConvLayer<scalar, fvPatchField, volMesh> ITHACAtorch::volScalarLayer

Definition at line 88 of file ConvLayer.H.

◆ volVectorLayer

typedef ConvLayer<vector, fvPatchField, volMesh> ITHACAtorch::volVectorLayer

Definition at line 89 of file ConvLayer.H.

Function Documentation

◆ isConst()

bool ITHACAtorch::isConst ( torch::Tensor & tTensor)

Definition at line 42 of file torchUTILITIES.C.

◆ load()

torch::Tensor ITHACAtorch::load ( const std::string fname)

Definition at line 70 of file torchUTILITIES.C.

◆ removeConstValues()

torch::Tensor ITHACAtorch::removeConstValues ( torch::Tensor input,
std::vector< int > & indices,
std::vector< double > & constValues )

Definition at line 35 of file torchUTILITIES.C.

◆ save()

void ITHACAtorch::save ( const torch::Tensor & torchTensor,
const std::string fname )

Definition at line 55 of file torchUTILITIES.C.