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

Version Path
tensor_stream-1.0.9 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.8 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.7 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.6 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.5 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.4 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.3 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.2 lib/tensor_stream/ops/log.rb
tensor_stream-1.0.1 lib/tensor_stream/ops/log.rb