Class: Tensorflow::Decorator::Function

Inherits:

Object

• Object
• Tensorflow::Decorator::Function

Defined in:

lib/tensorflow/decorators.rb

Instance Attribute Summary

• #input_signatures ⇒ Object readonly

  Returns the value of attribute input_signatures.

Instance Method Summary

• #initialize(input_signatures = []) ⇒ Function constructor

  A new instance of Function.

• #wrap(method) ⇒ Object

Constructor Details

#initialize(input_signatures = []) ⇒ Function

Returns a new instance of Function.

# File 'lib/tensorflow/decorators.rb', line 6

def initialize(input_signatures = [])
  @input_signatures = input_signatures
end

Instance Attribute Details

#input_signatures ⇒ Object (readonly)

Returns the value of attribute input_signatures.

# File 'lib/tensorflow/decorators.rb', line 4

def input_signatures
  @input_signatures
end

Instance Method Details

#wrap(method) ⇒ Object

# File 'lib/tensorflow/decorators.rb', line 10

def wrap(method)
  Graph::FunctionDef.new(method, self.input_signatures)
end