TensorFlow函数教程:tf.keras.backend.ndim

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
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部