Method: TensorStream::OpMaker#define_gradient

Defined in:
lib/tensor_stream/op_maker.rb

#define_gradient(&block) ⇒ Object 



117
118
119
 
# File 'lib/tensor_stream/op_maker.rb', line 117

def define_gradient(&block)
  @gradient = block
end