具有int输入的Keras层不能被bui

Question

我有一个复杂的keras模型，其中一个层是一个定制的预训练层，它需要“int32”作为输入。此模型作为继承自模型的类实现，其实现方式如下：

class MyModel(tf.keras.models.Model):

    def __init__(self, size, input_shape):
        super(MyModel, self).__init__()
        self.layer = My_Layer()
        self.build(input_shape)

    def call(self, inputs):
        return self.layer(inputs)

但是当它到达self.build方法时，会抛出下一个错误：

^{pr2}$

我怎样才能修好它？在