def __init__(self):
    super(Model, self).__init__()
    self.flatten = Flatten()
    self.dense1 = Dense(256, activation = 'relu')
    self.dense2 = Dense(128, activation = 'relu')
    self.dense3 = Dense(classes, activation = 'softmax')
    

  def call(self, x):
    print(x.shape)
    x = self.flatten(x)
    print(x.shape)
    x = self.dense1(x)
    print(x.shape)
    x = self.dense2(x)
    print(x.shape)
    y = self.dense3(x)
    print(y.shape)
    print(tf.matmul(x, self.w))
    return y

한층마다 가중치랑 편향 읽어오고 싶은데 아는사람?