Keras: input_shape函数

Posted 2021-02-21 zb-ml

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 
版权声明：本文为博主原创文章，转载请附上博文链接！