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