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

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

_action_scale_tensorMoose::LibtorchActionDistributionprotected
_data_typeMoose::LibtorchActionDistributionprotected
_device_typeMoose::LibtorchActionDistributionprotected
_log_std_parameterMoose::LibtorchGaussianActionDistributionprivate
_log_std_tensorMoose::LibtorchGaussianActionDistributionprivate
_meanMoose::LibtorchGaussianActionDistributionprivate
_mean_moduleMoose::LibtorchGaussianActionDistributionprivate
_nameMoose::LibtorchActionDistributionprotected
_num_inputsMoose::LibtorchActionDistributionprotected
_num_outputsMoose::LibtorchActionDistributionprotected
_output_scaling_factorsMoose::LibtorchActionDistributionprotected
_state_independent_stdMoose::LibtorchGaussianActionDistributionprivate
_std_moduleMoose::LibtorchGaussianActionDistributionprivate
_std_tensorMoose::LibtorchGaussianActionDistributionprivate
actionScaleTensor() constMoose::LibtorchActionDistributioninlineprotected
constructDistribution()Moose::LibtorchGaussianActionDistributionprivate
deterministicAction() const overrideMoose::LibtorchGaussianActionDistributionvirtual
entropy() const overrideMoose::LibtorchGaussianActionDistributionvirtual
initialize(c10::optional< at::Generator > generator=c10::nullopt) overrideMoose::LibtorchGaussianActionDistributionvirtual
isBounded() const overrideMoose::LibtorchGaussianActionDistributioninlinevirtual
LibtorchActionDistribution(const std::string &name, unsigned int num_inputs, unsigned int num_outputs, torch::DeviceType device_type=torch::kCPU, torch::ScalarType scalar_type=torch::kDouble, const std::vector< Real > &output_scaling_factors={})Moose::LibtorchActionDistribution
LibtorchGaussianActionDistribution(const std::string &name, unsigned int num_inputs, unsigned int num_outputs, torch::DeviceType device_type=torch::kCPU, torch::ScalarType scalar_type=torch::kDouble, bool build_on_construct=true, const std::vector< Real > &output_scaling_factors={}, bool state_independent_std=true)Moose::LibtorchGaussianActionDistribution
logProbability(const torch::Tensor &action) const overrideMoose::LibtorchGaussianActionDistributionvirtual
logStd()Moose::LibtorchGaussianActionDistributioninline
logStd() constMoose::LibtorchGaussianActionDistributioninline
meanModule()Moose::LibtorchGaussianActionDistributioninline
meanModule() constMoose::LibtorchGaussianActionDistributioninline
prepareAction(const torch::Tensor &action) constMoose::LibtorchActionDistributionprotected
prepareFeatures(const torch::Tensor &input) constMoose::LibtorchActionDistributionprotected
reset(const torch::Tensor &input) overrideMoose::LibtorchGaussianActionDistributionvirtual
sample(c10::optional< at::Generator > generator=c10::nullopt) const overrideMoose::LibtorchGaussianActionDistributionvirtual
stateIndependentStd() constMoose::LibtorchGaussianActionDistributioninline
stdModule()Moose::LibtorchGaussianActionDistributioninline
stdModule() constMoose::LibtorchGaussianActionDistributioninline
stdTensor() constMoose::LibtorchGaussianActionDistributioninline
synchronizeScalingFactorsFromBuffer()Moose::LibtorchActionDistribution