https://mooseframework.inl.gov
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
_nameMoose::LibtorchArtificialNeuralNetprotected
_num_inputsMoose::LibtorchArtificialNeuralNetprotected
_num_neurons_per_layerMoose::LibtorchArtificialNeuralNetprotected
_num_outputsMoose::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::LibtorchArtificialNeuralNet
dataType() constMoose::LibtorchArtificialNeuralNetinline
deviceType() constMoose::LibtorchArtificialNeuralNetinline
forward(const torch::Tensor &x) overrideMoose::LibtorchArtificialNeuralNetvirtual
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)Moose::LibtorchArtificialNeuralNet
LibtorchArtificialNeuralNet(const Moose::LibtorchArtificialNeuralNet &nn)Moose::LibtorchArtificialNeuralNet
name() constMoose::LibtorchArtificialNeuralNetinline
numHiddenLayers() constMoose::LibtorchArtificialNeuralNetinline
numInputs() constMoose::LibtorchArtificialNeuralNetinline
numNeuronsPerLayer() constMoose::LibtorchArtificialNeuralNetinline
numOutputs() constMoose::LibtorchArtificialNeuralNetinline
store(nlohmann::json &json) constMoose::LibtorchArtificialNeuralNet
~LibtorchNeuralNetBase()Moose::LibtorchNeuralNetBaseinlinevirtual