![]() |
IgANet
IGAnets - Isogeometric Analysis Networks
|
Public Member Functions | |
| bool | epoch (int64_t epoch) override |
| Initializes epoch. | |
| torch::Tensor | loss (const torch::Tensor &outputs, int64_t epoch) override |
| Computes the loss function. | |
Public Member Functions inherited from iganet::v1::IgANet< Optimizer, GeometryMap, Variable, IgABase > | |
| IgANet (const IgANetOptions &defaults={}, iganet::Options< typename Base::value_type > options=iganet::Options< typename Base::value_type >{}) | |
| Default constructor. | |
Private Types | |
| using | Base = iganet::v1::IgANet< Optimizer, GeometryMap, Variable > |
Private Attributes | |
| Base::variable_collPts_type | collPts_ |
Additional Inherited Members | |
Public Types inherited from iganet::v1::IgANet< Optimizer, GeometryMap, Variable, IgABase > | |
| using | Base = IgABase< GeometryMap, Variable > |
| Base type. | |
| using | optimizer_options_type = optimizer_options_type< Optimizer >::type |
| Type of the optimizer options. | |
| using | optimizer_type = Optimizer |
| Type of the optimizer. | |
Protected Attributes inherited from iganet::v1::IgANet< Optimizer, GeometryMap, Variable, IgABase > | |
| IgANetGenerator< typename Base::value_type > | net_ |
| IgANet generator. | |
| std::unique_ptr< optimizer_type > | opt_ |
| Optimizer. | |
| IgANetOptions | options_ |
| Options. | |
|
private |
|
inlineoverridevirtual |
Initializes epoch.
Implements iganet::v1::IgANet< Optimizer, GeometryMap, Variable, IgABase >.
|
inlineoverridevirtual |
Computes the loss function.
Implements iganet::v1::IgANet< Optimizer, GeometryMap, Variable, IgABase >.
|
private |