Keras: input_shape函數

keras.backend.int_shape(x)

返回張量或變量的尺寸，作為 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)

Numpy 實現

def int_shape(x):
    return x.shape

input_shape就是指輸入張量的shape。例如，input_dim=784，說明輸入是一個784維的向量，這相當於一個一階的張量，它的shape就是(784,)。因此，input_shape=(784,)。

input_dim = input_shape(input_dim,)

input_dim, input_length = input_shape(input_length, input_dim)

--------------------- 
作者：Flyver 
來源：CSDN 
原文：https://blog.csdn.net/x_ym/article/details/77728732 
版權聲明：本文為博主原創文章，轉載請附上博文鏈接！