A model which is nothing but a k-mode tensor of trainable parameters. Useful for making certain toy data sets.

normal_parameter_tensor.py

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

    def __init__(self, shape):
        super().__init__()
        self.w = tf.random.normal(shape=shape)
        self.w = tf.Variable(self.w, trainable=True)

    def call(self, x=None):
        return self.w