![]() |
IgANet
IgANets - Isogeometric Analysis Networks
|
This is the complete list of members for iganet::BatchNorm, including all inherited members.
apply(const torch::Tensor &input) const override | iganet::BatchNorm | inlinevirtual |
BatchNorm(const torch::Tensor &running_mean, const torch::Tensor &running_var, torch::nn::functional::BatchNormFuncOptions options={}) | iganet::BatchNorm | inlineexplicit |
BatchNorm(const torch::Tensor &running_mean, const torch::Tensor &running_var, const torch::Tensor &weight, const torch::Tensor &bias, double eps, double momentum, bool training=false) | iganet::BatchNorm | inlineexplicit |
name() const noexcept | iganet::utils::FullQualifiedName | inlineprotectedvirtual |
name_ | iganet::utils::FullQualifiedName | mutableprotected |
options() const | iganet::BatchNorm | inline |
options() | iganet::BatchNorm | inline |
options_ | iganet::BatchNorm | private |
pretty_print(std::ostream &os=Log(log::info)) const noexcept override | iganet::BatchNorm | inlinevirtual |
read(torch::serialize::InputArchive &archive, const std::string &key="batch_norm") override | iganet::BatchNorm | inlinevirtual |
running_mean() const | iganet::BatchNorm | inline |
running_mean() | iganet::BatchNorm | inline |
running_mean_ | iganet::BatchNorm | private |
running_var() const | iganet::BatchNorm | inline |
running_var() | iganet::BatchNorm | inline |
running_var_ | iganet::BatchNorm | private |
write(torch::serialize::OutputArchive &archive, const std::string &key="batch_norm") const override | iganet::BatchNorm | inlinevirtual |
~ActivationFunction()=default | iganet::ActivationFunction | virtual |
~BatchNorm() override=default | iganet::BatchNorm |