Tensorflow2 Layer with an object containing trainable variable

Question

import tensorflow as tf
from tensorflow.keras.layers import Dense, Conv2D
from tensorflow.keras import backend as K

class Bias(tf.Module):

    def __init__(self, units