Method: TensorStream::OpStub#expand_dims

Defined in:
lib/tensor_stream/generated_stub/ops.rb

#expand_dims(input, axis, name: nil) ⇒ Object

Inserts a dimension of 1 into a tensor’s shape. Given a tensor input, this operation inserts a dimension of 1 at the dimension index axis of input’s shape. The dimension index axis starts at zero; if you specify a negative number for axis it is counted backward from the end.

Options:

Options Hash (name:):

  • Optional (Object) —

    name



138
139
140
 
# File 'lib/tensor_stream/generated_stub/ops.rb', line 138

def expand_dims(input, axis, name: nil)
  _op(:expand_dims, input, axis, name: name)
end