Hello , i have a Qnode with data-reupload technique in which i use strongly entangled layers but i would like to use basicentangler layers instead and for some reason i get some errors

```
> @qml.qnode(dev, interface="tf", grad_method="backprop")
> def qnode(inputs, weights):
> for i in range(blocks):
> qml.templates.AngleEmbedding(inputs, wires=range(n_qubits))
> # qml.templates.BasicEntanglerLayers(weights, wires=range(n_qubits)) #BASIC ENTANGLER LAYERS , chan
> qml.templates.StronglyEntanglingLayers(weights[i], wires=range(n_qubits)) #STRONGLY ENTANGLING LAYERS
> return [qml.expval(qml.PauliZ(i)) for i in range(n_qubits)]
>
>
> weights_shape = (blocks, layers, n_qubits, 3)
>
>
>
weight_shapes = {"weights": weights_shape}
qlayer = qml.qnn.KerasLayer(qnode, weight_shapes, output_dim=n_qubits)
```

This is the working code with strongly entangling layers in which every â€śblockâ€ť must be initialized with random weights (which hopefully works as i describe it). Can you let me know what changes to make so it works with basic entangler layers ? thanks.