Keras: input_shape函数

xiaoxiaokeke 2020-04-26

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

相关推荐