Watch keynotes, product sessions, workshops, and more from Google I/O See playlist

tf.rank

Returns the rank of a tensor.

Used in the notebooks

Used in the tutorials

See also tf.shape.

Returns a 0-D int32 Tensor representing the rank of input.

For example:

# shape of tensor 't' is [2, 2, 3]
t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.rank(t)  # 3

input A Tensor or SparseTensor.
name A name for the operation (optional).

A Tensor of type int32.

numpy compatibility

Equivalent to np.ndim