阅读(8.9k) 书签 (0)

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

2019-03-13 17:34 更新

tf.keras.backend.int_shape函数

tf.keras.backend.int_shape(x)

定义在:tensorflow/python/keras/backend.py。

返回张量或变量的shape,作为int或None条目的元组。 

参数:

  • x:张量或变量。

返回:

整数元组(或 None 条目)

示例:

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