tf-keras-skeleton.py 文件源码

def build(self, input_shape):
            input_shape = tensor_shape.TensorShape(input_shape).as_list()
            param_shape = input_shape[1:]
            self.param_broadcast = [False] * len(param_shape)
            if self.shared_axes is not None:
              for i in self.shared_axes:
                param_shape[i - 1] = 1
                self.param_broadcast[i - 1] = True
            self.alpha = self.add_weight(
                shape=param_shape,
                name='alpha',
                initializer=self.alpha_initializer,
                regularizer=self.alpha_regularizer,
                constraint=self.alpha_constraint)
            # Set input spec
            axes = {}
            if self.shared_axes:
              for i in range(1, len(input_shape)):
                if i not in self.shared_axes:
                  axes[i] = input_shape[i]
            self.input_spec = InputSpec(ndim=len(input_shape), axes=axes)
            self.built = True