tf-keras-skeleton.py 文件源码

def count_params(self):
            """Count the total number of scalars composing the weights.

            Returns:
                An integer count.

            Raises:
                RuntimeError: if the layer isn't yet built
                    (in which case its weights aren't yet defined).
            """
            if not self.built:
              if self.__class__.__name__ == 'Sequential':
                self.build()  # pylint: disable=no-value-for-parameter
              else:
                raise RuntimeError('You tried to call `count_params` on ' + self.name +
                                   ', but the layer isn\'t built. '
                                   'You can build it manually via: `' + self.name +
                                   '.build(batch_input_shape)`.')
            return sum([K.count_params(p) for p in self.weights])