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

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

_action_scale_tensorMoose::LibtorchActionDistributionprotected
_data_typeMoose::LibtorchActionDistributionprotected
_device_typeMoose::LibtorchActionDistributionprotected
_nameMoose::LibtorchActionDistributionprotected
_num_inputsMoose::LibtorchActionDistributionprotected
_num_outputsMoose::LibtorchActionDistributionprotected
_output_scaling_factorsMoose::LibtorchActionDistributionprotected
actionScaleTensor() constMoose::LibtorchActionDistributioninlineprotected
deterministicAction() const =0Moose::LibtorchActionDistributionpure virtual
entropy() const =0Moose::LibtorchActionDistributionpure virtual
initialize(c10::optional< at::Generator > generator=c10::nullopt)=0Moose::LibtorchActionDistributionpure virtual
isBounded() const =0Moose::LibtorchActionDistributionpure virtual
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
logProbability(const torch::Tensor &action) const =0Moose::LibtorchActionDistributionpure virtual
prepareAction(const torch::Tensor &action) constMoose::LibtorchActionDistributionprotected
prepareFeatures(const torch::Tensor &input) constMoose::LibtorchActionDistributionprotected
reset(const torch::Tensor &input)=0Moose::LibtorchActionDistributionpure virtual
sample(c10::optional< at::Generator > generator=c10::nullopt) const =0Moose::LibtorchActionDistributionpure virtual
synchronizeScalingFactorsFromBuffer()Moose::LibtorchActionDistribution