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

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

_action_scale_tensorMoose::LibtorchActionDistributionprotected
_alpha_beta_tensorMoose::LibtorchBetaActionDistributionprivate
_alpha_moduleMoose::LibtorchBetaActionDistributionprivate
_alpha_tensorMoose::LibtorchBetaActionDistributionprivate
_beta_moduleMoose::LibtorchBetaActionDistributionprivate
_beta_tensorMoose::LibtorchBetaActionDistributionprivate
_data_typeMoose::LibtorchActionDistributionprotected
_device_typeMoose::LibtorchActionDistributionprotected
_log_normMoose::LibtorchBetaActionDistributionprivate
_max_tensorMoose::LibtorchBetaActionDistributionprivate
_maximum_valuesMoose::LibtorchBetaActionDistributionprivate
_meanMoose::LibtorchBetaActionDistributionprivate
_min_tensorMoose::LibtorchBetaActionDistributionprivate
_minimum_valuesMoose::LibtorchBetaActionDistributionprivate
_nameMoose::LibtorchActionDistributionprotected
_num_inputsMoose::LibtorchActionDistributionprotected
_num_outputsMoose::LibtorchActionDistributionprotected
_output_scaling_factorsMoose::LibtorchActionDistributionprotected
actionScaleTensor() constMoose::LibtorchActionDistributioninlineprotected
alphaModule()Moose::LibtorchBetaActionDistributioninline
alphaModule() constMoose::LibtorchBetaActionDistributioninline
alphaTensor() constMoose::LibtorchBetaActionDistributioninline
betaModule()Moose::LibtorchBetaActionDistributioninline
betaModule() constMoose::LibtorchBetaActionDistributioninline
betaTensor() constMoose::LibtorchBetaActionDistributioninline
constructDistribution()Moose::LibtorchBetaActionDistributionprivate
deterministicAction() const overrideMoose::LibtorchBetaActionDistributionvirtual
entropy() const overrideMoose::LibtorchBetaActionDistributionvirtual
initialize(c10::optional< at::Generator > generator=c10::nullopt) overrideMoose::LibtorchBetaActionDistributionvirtual
isBounded() const overrideMoose::LibtorchBetaActionDistributioninlinevirtual
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
LibtorchBetaActionDistribution(const std::string &name, unsigned int num_inputs, unsigned int num_outputs, const std::vector< Real > &minimum_values, const std::vector< Real > &maximum_values, torch::DeviceType device_type=torch::kCPU, torch::ScalarType scalar_type=torch::kDouble, bool build_on_construct=true, const std::vector< Real > &output_scaling_factors={})Moose::LibtorchBetaActionDistribution
logProbability(const torch::Tensor &action) const overrideMoose::LibtorchBetaActionDistributionvirtual
prepareAction(const torch::Tensor &action) constMoose::LibtorchActionDistributionprotected
prepareFeatures(const torch::Tensor &input) constMoose::LibtorchActionDistributionprotected
reset(const torch::Tensor &input) overrideMoose::LibtorchBetaActionDistributionvirtual
sample(c10::optional< at::Generator > generator=c10::nullopt) const overrideMoose::LibtorchBetaActionDistributionvirtual
synchronizeScalingFactorsFromBuffer()Moose::LibtorchActionDistribution