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

2 entries across 2 versions & 1 rubygems

Version	Path
tensor_stream-0.1.1	lib/tensor_stream/nn/nn_ops.rb
tensor_stream-0.1.0	lib/tensor_stream/nn/nn_ops.rb