Struct GradientsParams 

pub struct GradientsParams { /* private fields */ }

Data type that contains gradients for parameters.

Implementations

impl GradientsParams

pub fn new() -> Self

Creates a new GradientsParams.

pub fn from_grads<M: AutodiffModule>(grads: Gradients, module: &M) -> Self

Extract each tensor gradients for the given module.

Note: This consumes the gradients. See [‘from_module’] to extract gradients only for a specific module.

pub fn from_module<M: AutodiffModule>(grads: &mut Gradients, module: &M) -> Self

Extract each tensor gradients for the given module.

pub fn from_params<M: AutodiffModule>( grads: &mut Gradients, module: &M, params: &[ParamId], ) -> Self

Extract tensor gradients for the given module and given parameters.

pub fn get<const D: usize>(&self, id: ParamId) -> Option<Tensor<D>>

Get the gradients for the given parameter id.

Notes

You should use remove if you want to get the gradients only one time.

pub fn remove<const D: usize>(&mut self, id: ParamId) -> Option<Tensor<D>>

Remove the gradients for the given parameter id.

pub fn register<const D: usize>(&mut self, id: ParamId, value: Tensor<D>)

Register a gradients tensor for the given parameter id.

Notes

If a tensor is already registered for the given parameter id, it will be replaced.

pub fn len(&self) -> usize

The number of gradients tensors registered.

pub fn is_empty(&self) -> bool

If any tensor is contained.

pub fn to_device<M: AutodiffModule>(self, device: &Device, module: &M) -> Self

Change the device of each tensor gradients registered for the given module.

Trait Implementations

impl Debug for GradientsParams

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for GradientsParams

fn default() -> GradientsParams

Returns the “default value” for a type.

impl From<GradientsParams> for GradAdaptor

fn from(grads: GradientsParams) -> Self

Converts to this type from the input type.