Struct NNOptimizerRmsProp 

pub struct NNOptimizerRmsProp { /* private fields */ }

Wraps MPSNNOptimizerRMSProp.

Implementations

impl NNOptimizerRmsProp

pub const fn as_ptr(&self) -> *mut c_void

Returns the retained Objective-C pointer backing this wrapper.

impl NNOptimizerRmsProp

pub fn learning_rate(&self) -> f32

Wraps the corresponding Metal Performance Shaders method.

pub fn set_learning_rate(&self, value: f32)

Wraps the corresponding Metal Performance Shaders method.

pub fn gradient_rescale(&self) -> f32

Wraps the corresponding Metal Performance Shaders method.

pub fn apply_gradient_clipping(&self) -> bool

Wraps the corresponding Metal Performance Shaders method.

pub fn set_apply_gradient_clipping(&self, value: bool)

Wraps the corresponding Metal Performance Shaders method.

pub fn gradient_clip_max(&self) -> f32

Wraps the corresponding Metal Performance Shaders method.

pub fn gradient_clip_min(&self) -> f32

Wraps the corresponding Metal Performance Shaders method.

pub fn regularization_scale(&self) -> f32

Wraps the corresponding Metal Performance Shaders method.

pub fn regularization_type(&self) -> usize

Wraps the corresponding Metal Performance Shaders method.

impl NNOptimizerRmsProp

pub fn new(device: &MetalDevice, learning_rate: f32) -> Option<Self>

Wraps a constructor on MPSNNOptimizerRMSProp.

pub fn new_with_options( device: &MetalDevice, decay: f64, epsilon: f32, optimizer_descriptor: &NNOptimizerDescriptor, ) -> Option<Self>

Wraps a constructor on MPSNNOptimizerRMSProp.

pub fn as_optimizer(&self) -> Option<NNOptimizer>

Wraps the corresponding MPSNNOptimizerRMSProp conversion helper.

pub fn decay(&self) -> f64

Wraps the corresponding MPSNNOptimizerRMSProp method.

pub fn epsilon(&self) -> f32

Wraps the corresponding MPSNNOptimizerRMSProp method.

pub fn encode_vector( &self, command_buffer: &CommandBuffer, input_gradient_vector: &Vector, input_values_vector: &Vector, input_sum_of_squares_vector: &Vector, result_values_vector: &Vector, )

Wraps the corresponding MPSNNOptimizerRMSProp encode entry point.

pub fn encode_matrix( &self, command_buffer: &CommandBuffer, input_gradient_matrix: &Matrix, input_values_matrix: &Matrix, input_sum_of_squares_matrix: &Matrix, result_values_matrix: &Matrix, )

Wraps the corresponding MPSNNOptimizerRMSProp encode entry point.

Trait Implementations

impl Drop for NNOptimizerRmsProp

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.

impl Send for NNOptimizerRmsProp

impl Sync for NNOptimizerRmsProp