码迷,mamicode.com
首页 > 其他好文 > 详细

Keras: input_shape函数

时间:2020-04-26 18:35:13      阅读:361      评论:0      收藏:0      [点我收藏+]

标签:turn   元组   tail   inpu   def   int   例子   shape   numpy   

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函数

标签:turn   元组   tail   inpu   def   int   例子   shape   numpy   

原文地址:https://www.cnblogs.com/zb-ml/p/12781346.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!