https://mooseframework.inl.gov
Loading...
Searching...
No Matches
Moose::LibtorchArtificialNeuralNet Member List

This is the complete list of members for Moose::LibtorchArtificialNeuralNet, including all inherited members.

_activation_functionMoose::LibtorchArtificialNeuralNetprotected
_data_typeMoose::LibtorchArtificialNeuralNetprotected
_device_typeMoose::LibtorchArtificialNeuralNetprotected
_input_scale_tensorMoose::LibtorchArtificialNeuralNetprotected
_input_scaling_factorsMoose::LibtorchArtificialNeuralNetprotected
_input_shift_factorsMoose::LibtorchArtificialNeuralNetprotected
_input_shift_tensorMoose::LibtorchArtificialNeuralNetprotected
_nameMoose::LibtorchArtificialNeuralNetprotected
_num_inputsMoose::LibtorchArtificialNeuralNetprotected
_num_neurons_per_layerMoose::LibtorchArtificialNeuralNetprotected
_num_outputsMoose::LibtorchArtificialNeuralNetprotected
_output_scale_tensorMoose::LibtorchArtificialNeuralNetprotected
_output_scaling_factorsMoose::LibtorchArtificialNeuralNetprotected
_weightsMoose::LibtorchArtificialNeuralNetprotected
activationFunctions() constMoose::LibtorchArtificialNeuralNetinline
addLayer(const std::string &layer_name, const std::unordered_map< std::string, unsigned int > &parameters)Moose::LibtorchArtificialNeuralNetvirtual
constructNeuralNetwork()Moose::LibtorchArtificialNeuralNetvirtual
dataType() constMoose::LibtorchArtificialNeuralNetinline
determineGain(const std::string &activation)Moose::LibtorchArtificialNeuralNet
deviceType() constMoose::LibtorchArtificialNeuralNetinline
forward(const torch::Tensor &x) overrideMoose::LibtorchArtificialNeuralNetvirtual
initializeAffineBuffers()Moose::LibtorchArtificialNeuralNetprotected
initializeNeuralNetwork(c10::optional< at::Generator > generator=c10::nullopt)Moose::LibtorchArtificialNeuralNetvirtual
inputScalingFactors() constMoose::LibtorchArtificialNeuralNetinline
inputShiftFactors() constMoose::LibtorchArtificialNeuralNetinline
LibtorchArtificialNeuralNet(const std::string name, const unsigned int num_inputs, const unsigned int num_outputs, const std::vector< unsigned int > &num_neurons_per_layer, const std::vector< std::string > &activation_function={"relu"}, const torch::DeviceType device_type=torch::kCPU, const torch::ScalarType scalar_type=torch::kDouble, const bool build_on_construct=true, const std::vector< Real > &input_shift_factors={}, const std::vector< Real > &input_scaling_factors={}, const std::vector< Real > &output_scaling_factors={})Moose::LibtorchArtificialNeuralNet
LibtorchArtificialNeuralNet(const Moose::LibtorchArtificialNeuralNet &nn, const bool build_on_construct=true)Moose::LibtorchArtificialNeuralNet
name() constMoose::LibtorchArtificialNeuralNetinline
numHiddenLayers() constMoose::LibtorchArtificialNeuralNetinline
numInputs() constMoose::LibtorchArtificialNeuralNetinline
numNeuronsPerLayer() constMoose::LibtorchArtificialNeuralNetinline
numOutputs() constMoose::LibtorchArtificialNeuralNetinline
outputScalingFactors() constMoose::LibtorchArtificialNeuralNetinline
preprocessInput(const torch::Tensor &x) constMoose::LibtorchArtificialNeuralNetprotectedvirtual
scaleOutput(const torch::Tensor &y) constMoose::LibtorchArtificialNeuralNetprotectedvirtual
setAffineFactors(const std::vector< Real > &factors, unsigned int expected_size, Real default_value, const std::string &factor_name)Moose::LibtorchArtificialNeuralNetprotectedstatic
store(nlohmann::json &json) constMoose::LibtorchArtificialNeuralNet
synchronizeAffineFactorsFromBuffers()Moose::LibtorchArtificialNeuralNet
~LibtorchNeuralNetBase()Moose::LibtorchNeuralNetBaseinlinevirtual