Contents

module TensorStream
  # High level machine learning functions
  class NN
    def self.softmax(logits, _options = {})
      TensorStream.exp(logits) / TensorStream.reduce_sum(TensorStream.exp(logits))
    end

    def self.relu(features, name: nil)
      TensorStream.max(features, 0, name: "relu_#{name}")
    end
  end

  # tensorflow compatibility
  def self.nn
    TensorStream::NN
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
tensor_stream-0.1.4	lib/tensor_stream/nn/nn_ops.rb
tensor_stream-0.1.3	lib/tensor_stream/nn/nn_ops.rb
tensor_stream-0.1.2	lib/tensor_stream/nn/nn_ops.rb