Method: Tensorflow::Train::Optimizer#apply_gradients

Defined in:
lib/tensorflow/train/optimizer.rb

#apply_gradients(grads_and_vars, global_step: nil, name: nil) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# File 'lib/tensorflow/train/optimizer.rb', line 28

def apply_gradients(grads_and_vars, global_step: nil, name: nil)
  varlist = grads_and_vars.map { |_grad, var| var }
  #create_slots(varlist)

  #TensorStream.name_scope(name, default: @name) do

    prepare
    apply_ops = grads_and_vars.map do |grad, var|
      #TensorStream.name_scope("update_" + var.op.name) do

        apply_dense(grad, var)
      #end

    end

    if global_step.nil?
      finish(apply_ops, name)
    else
      global_step.handle.graph.control_dependencies([finish(apply_ops, "update")]) do
        global_step.assign_add(Tensorflow.constant(1, dtype:global_step.dtype))
      end
    end
  #end

end