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 ConvLayer<scalar, fvsPatchField, surfaceMesh> ITHACAtorch::surfaceScalarLayer |
Definition at line 90 of file ConvLayer.H.
typedef ConvLayer<scalar, fvPatchField, volMesh> ITHACAtorch::volScalarLayer |
Definition at line 88 of file ConvLayer.H.
typedef ConvLayer<vector, fvPatchField, volMesh> ITHACAtorch::volVectorLayer |
Definition at line 89 of file ConvLayer.H.
bool ITHACAtorch::isConst | ( | torch::Tensor & | tTensor | ) |
Definition at line 42 of file torchUTILITIES.C.
torch::Tensor ITHACAtorch::load | ( | const std::string | fname | ) |
Definition at line 70 of file torchUTILITIES.C.
torch::Tensor ITHACAtorch::removeConstValues | ( | torch::Tensor | input, |
std::vector< int > & | indices, | ||
std::vector< double > & | constValues ) |
Definition at line 35 of file torchUTILITIES.C.
void ITHACAtorch::save | ( | const torch::Tensor & | torchTensor, |
const std::string | fname ) |
Definition at line 55 of file torchUTILITIES.C.