class TensorStream::Train::GradientDescentOptimizer
High Level implementation of the gradient descent algorithm
Attributes
learning_rate[RW]
Public Class Methods
new(learning_rate, use_locking: false, name: "GradientDescent")
click to toggle source
Calls superclass method
# File lib/tensor_stream/train/gradient_descent_optimizer.rb, line 9 def initialize(learning_rate, use_locking: false, name: "GradientDescent") @learning_rate = learning_rate @learning_rate_tensor = nil super(name: name, use_locking: use_locking) end
Protected Instance Methods
apply_dense(grad, var)
click to toggle source
# File lib/tensor_stream/train/gradient_descent_optimizer.rb, line 22 def apply_dense(grad, var) i_op(:apply_gradient_descent, var, TensorStream.cast(@learning_rate_tensor, grad.data_type), grad) end
prepare()
click to toggle source
# File lib/tensor_stream/train/gradient_descent_optimizer.rb, line 17 def prepare learning_rate = call_if_callable(@learning_rate) @learning_rate_tensor = convert_to_tensor(learning_rate, name: "learning_rate") end