module TensorFlow

Enhance the module's Ruby API.

Public Class Methods

session(*args) { |sess| ... } click to toggle source

Pass a tf.Session() to a block and auto-close it.

# File lib/tensorflow.rb, line 9
def self.session(*args, &block)
  sess = self.Session.new(*args)
  return(sess) unless block
  begin; yield(sess); ensure; sess.close; end
end