tensorflow 自作layerの作成

Pythonにて自作レイヤーを作成し,モデルを作りたいと考えています.
特に,下の例で太字の部分を自作関数で作りたいと考えています.

例)

x -> [1,x,...,x^n] -> Activation(Σ w_i x^i) -> ...-> y

そして,以下のように定義てmodelを

python
1class MyDenseLayer(tf.keras.layers.Layer):
2    def __init__(self, num_outputs):
3        super().__init__()
4        self.num_outputs = num_outputs
5    
6    def call(self, input):
7        return tf.Variable(list( input **i for i in range(self.num_outputs)))
8
9class MyModel(tf.keras.Model):
10    def __init__(self):
11        tf.random.set_seed(2021)
12        super().__init__()
13        self.dense1 = MyDenseLayer(10)
14        self.dense2 = tf.keras.layers.Dense(3, activation=tf.nn.tanh)
15        self.dense3 = tf.keras.layers.Dense(3, activation=tf.nn.tanh)
16        self.dense4 = tf.keras.layers.Dense(1)
17    
18    def call(self, inputs, training=False):
19        x = self.dense1(inputs)
20        x = self.dense2(x)
21        x = self.dense3(x)
22        return self.dense4(x)

と定義し,以下を実装しました.

model=MyModel()
model.build(input_shape=(None,1))
model.summary()

そして,以下のようなエラーが起きてしまいました.

ValueError: You cannot build your model by calling `build` if your layers do not support float type inputs. Instead, in order to instantiate and build your model, `call` your model on real tensor data (of the correct dtype).

おそらく,MyDenseLayerの部分にbuild関数を定義したり,する必要があると思いますが,
どのようにしたら良いかわからないため,
どのように訂正したら良いかご教授いただけると幸いです.

また, call関数においてのタイプはtensorflow.python.ops.resource_variable_ops.ResourceVariable で問題ないかどうかもご教授していただけると幸いです.