Sha256: a16814c805379931db072a79457349c82bdec489e125a7b3862c87aed633aec2
Contents?: true
Size: 365 Bytes
Versions: 9
Compression:
Stored size: 365 Bytes
Contents
TensorStream::OpMaker.define_operation :log do |op| op.what_it_does "Computes natural logarithm of x element-wise." op.parameter :input, "tensor X" op.option :name, "Optional name", :nil op.define_gradient do |grad, node, params| grad * TensorStream.reciprocal(params[0]) end op.define_shape do |tensor| tensor.inputs[0].shape.shape end end
Version data entries
9 entries across 9 versions & 1 rubygems