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
版权声明:本文为博主原创文章,转载请附上博文链接!
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Keras: input_shape函数 - Python技术站