In this article, you will see how to get the dimension of a tensor in TensorFlow. First, import the TensorFlow framework, then create a tensor and use the tensor.ndim function to find the dimension of a tensor. I highly recommend you get this book, “Deep Learning with TensorFlow 2 and Keras” to learn Deep Learning.
# Import TensorFlow import tensorflow as tf # Create a tensor a = tf.constant([[5,10], [30,70]]) # Display dimension of tensor print("The dimension of a Tensor is", a.ndim)
The dimension of a Tensor is 2