https://mooseframework.inl.gov
Loading...
Searching...
No Matches
Classes | Namespaces | Functions
LibtorchActorNeuralNet.h File Reference

Go to the source code of this file.

Classes

class  Moose::LibtorchActorNeuralNet
 Feed-forward actor network coupled to a Gaussian or Beta action distribution. More...
 

Namespaces

namespace  Moose
 

Functions

void Moose::to_json (nlohmann::json &json, const Moose::LibtorchActorNeuralNet *const &network)
 Dump an actor network into JSON for reporter output and debugging.
 
void Moose::loadLibtorchActorNeuralNetState (Moose::LibtorchActorNeuralNet &nn, const std::string &filename)
 Load an actor checkpoint written as a native libtorch state archive.
 
template<>
void dataStore< Moose::LibtorchActorNeuralNet > (std::ostream &stream, std::shared_ptr< Moose::LibtorchActorNeuralNet > &nn, void *context)
 Serialize the actor-network metadata needed for restart.
 
template<>
void dataLoad< Moose::LibtorchActorNeuralNet > (std::istream &stream, std::shared_ptr< Moose::LibtorchActorNeuralNet > &nn, void *context)
 Deserialize the actor-network metadata needed for restart.
 
template<>
void dataStore< Moose::LibtorchActorNeuralNet const > (std::ostream &stream, Moose::LibtorchActorNeuralNet const *&nn, void *context)
 Placeholder serializer for reporter-only actor pointers.
 
template<>
void dataLoad< Moose::LibtorchActorNeuralNet const > (std::istream &stream, Moose::LibtorchActorNeuralNet const *&nn, void *context)
 Placeholder deserializer for reporter-only actor pointers.
 

Function Documentation

◆ dataLoad< Moose::LibtorchActorNeuralNet >()

template<>
void dataLoad< Moose::LibtorchActorNeuralNet > ( std::istream &  stream,
std::shared_ptr< Moose::LibtorchActorNeuralNet > &  nn,
void *  context 
)

Deserialize the actor-network metadata needed for restart.

Parameters
streamStream that provides the serialized data.
nnActor network shared pointer to populate.
contextSerialization context passed through MOOSE data I/O.

◆ dataLoad< Moose::LibtorchActorNeuralNet const >()

template<>
void dataLoad< Moose::LibtorchActorNeuralNet const > ( std::istream &  stream,
Moose::LibtorchActorNeuralNet const *&  nn,
void *  context 
)

Placeholder deserializer for reporter-only actor pointers.

Parameters
streamStream that would provide the serialized data.
nnReporter actor pointer.
contextSerialization context passed through MOOSE data I/O.

◆ dataStore< Moose::LibtorchActorNeuralNet >()

template<>
void dataStore< Moose::LibtorchActorNeuralNet > ( std::ostream &  stream,
std::shared_ptr< Moose::LibtorchActorNeuralNet > &  nn,
void *  context 
)

Serialize the actor-network metadata needed for restart.

Parameters
streamStream that receives the serialized data.
nnActor network shared pointer to serialize.
contextSerialization context passed through MOOSE data I/O.

◆ dataStore< Moose::LibtorchActorNeuralNet const >()

template<>
void dataStore< Moose::LibtorchActorNeuralNet const > ( std::ostream &  stream,
Moose::LibtorchActorNeuralNet const *&  nn,
void *  context 
)

Placeholder serializer for reporter-only actor pointers.

Parameters
streamStream that would receive the serialized data.
nnReporter actor pointer.
contextSerialization context passed through MOOSE data I/O.