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