Method: TensorStream::Train::SlotCreator#create_slot_var
- Defined in:
- lib/tensor_stream/train/slot_creator.rb
#create_slot_var(_primary, val, scope, shape) ⇒ Object
Helper function for creating a slot variable.
8 9 10 11 12 |
# File 'lib/tensor_stream/train/slot_creator.rb', line 8 def create_slot_var(_primary, val, scope, shape) slot = get_variable(scope, initializer: val, trainable: false, shape: shape, validate_shape: val.shape && val.shape.known?) slot end |