TensorFlow函數(shù)教程：tf.keras.backend.ndim

tf.keras.backend.ndim函數(shù)

tf.keras.backend.ndim(x)

定義在：tensorflow/python/keras/backend.py。

以整數(shù)形式返回張量中的軸數(shù)。

參數(shù)：

• x：張量或變量。

返回：

整數(shù) (標(biāo)量)，軸的數(shù)量。

示例：

>>> from keras import backend as K
>>> inputs = K.placeholder(shape=(2, 4, 5))
>>> val = np.array([[1, 2], [3, 4]])
>>> kvar = K.variable(value=val)
>>> K.ndim(inputs)
3
>>> K.ndim(kvar)
2