Class: TensorStream::NN

Inherits:

Object

• Object
• TensorStream::NN

Defined in:

lib/tensor_stream/nn/nn_ops.rb

Overview

High level machine learning functions

Class Method Summary

• .relu(features, name: nil) ⇒ Object
• .softmax(logits, _options = {}) ⇒ Object

Class Method Details

.relu(features, name: nil) ⇒ Object

# File 'lib/tensor_stream/nn/nn_ops.rb', line 8

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

.softmax(logits, _options = {}) ⇒ Object

# File 'lib/tensor_stream/nn/nn_ops.rb', line 4

def self.softmax(logits, _options = {})
  TensorStream.exp(logits) / TensorStream.reduce_sum(TensorStream.exp(logits))
end