import tensorflow as tf import tensorflow.keras.layers as nn import numpy as np class Base(tf.keras.Model): def __init__(self): super(Base, self).__init__()