TensorFlow 2 version | View source on GitHub |
Adds a 1-sized dimension at index "axis".
tf.keras.backend.expand_dims(
x, axis=-1
)
Arguments | |
---|---|
x
|
A tensor or variable. |
axis
|
Position where to add a new axis. |
Returns | |
---|---|
A tensor with expanded dimensions. |