Kerasで入力にバイアスを加えるだけの層を作りたい

Question

Kerasで入力にバイアスを加えるだけの層が作りたいです。
入力サイズが4*4*512で、バイアスのサイズは512で、バイアスは学習する変数のようなものにしたいです。

x = Input([4,4,512],name="input") # 入力
# biasを作成
# biasを入力に加える

というようなコードが書きたいです。
よろしくお願いします。

---

追記

class MyAddBias(Layer):
    def __init__(self, input_shape, **kwargs):
        self.output_dim = output_dim
        bias_shape = input_shape[2]
        super(MyAddBias, self).__init__(**kwargs)

    def build(self):
        self.bias   = self.add_weight(name='bias',
                                      shape=(1, bias_shape),
                                      initializer='zeros',
                                      trainable=True)        
        super(MyAddBias, self).build(input_shape)

    def call(self, x):
        return x + self.bias
    
    def compute_output_shape(self, input_shape):
        return(input_shape[0], self.output_dim)

こんな感じで書いてみましたがTypeError: __int__ returned non-int (type NoneType)というエラーが出ます