tf-keras-skeleton.py 文件源码

python
阅读 16 收藏 0 点赞 0 评论 0

项目:LIE 作者: EmbraceLife 项目源码 文件源码
def count_params(x):
      """Returns the number of scalars in a Keras variable.

      Arguments:
          x: Keras variable.

      Returns:
          Integer, the number of scalars in `x`.

      Example:
      ```python
          >>> kvar = K.zeros((2,3))
          >>> K.count_params(kvar)
          6
          >>> K.eval(kvar)
          array([[ 0.,  0.,  0.],
                 [ 0.,  0.,  0.]], dtype=float32)
"""
  shape = x.get_shape()
  return np.prod([shape[i]._value for i in range(len(shape))])

```

评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号