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.
1.13.2