TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning neural networks.
expand_dims() is used to insert an addition dimension in input Tensor.
Syntax: tensorflow.expand_dims( input, axis, name)
- input: It is the input Tensor.
- axis: It defines the index at which dimension should be inserted. If input has D dimensions then axis must have value in range [-(D+1), D].
- name(optional): It defines the name for the operation.
Returns: It returns a Tensor with expanded dimension.
x: tf.Tensor( [[ 2 3 6] [ 4 8 15]], shape=(2, 3), dtype=int32) res: tf.Tensor( [[[ 2 3 6]] [[ 4 8 15]]], shape=(2, 1, 3), dtype=int32) # shape has chaged from (2, 3) to (2, 1, 3)
x: tf.Tensor( [[ 2 3 6] [ 4 8 15]], shape=(2, 3), dtype=int32) res: tf.Tensor( [[[ 2 3 6] [ 4 8 15]]], shape=(1, 2, 3), dtype=int32) # shape has chaged from (2, 3) to (1, 2, 3)
- Important differences between Python 2.x and Python 3.x with examples
- Python | Set 4 (Dictionary, Keywords in Python)
- Python | Sort Python Dictionaries by Key or Value
- Python | Merge Python key values to list
- Reading Python File-Like Objects from C | Python
- Python | Add Logging to a Python Script
- Python | Add Logging to Python Libraries
- Python | Visualizing O(n) using Python
- Python | Index of Non-Zero elements in Python list
- Python | Convert list to Python array
- MySQL-Connector-Python module in Python
- Python - Read blob object in python using wand library
- Python | PRAW - Python Reddit API Wrapper
- twitter-text-python (ttp) module - Python
- Reusable piece of python functionality for wrapping arbitrary blocks of code : Python Context Managers
- Python program to check if the list contains three consecutive common numbers in Python
- Creating and updating PowerPoint Presentations in Python using python - pptx
- How to write an empty function in Python - pass statement?
- Operator Functions in Python | Set 2
If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to email@example.com. See your article appearing on the GeeksforGeeks main page and help other Geeks.
Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.