TensorFlow函数教程:tf.keras.backend.ndim
2019-03-20 18:19 更新
tf.keras.backend.ndim函数
tf.keras.backend.ndim(x)
定义在:tensorflow/python/keras/backend.py。
以整数形式返回张量中的轴数。
参数:
- x:张量或变量。
返回:
整数 (标量),轴的数量。
示例:
>>> 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