问如何在Keras中使用张量板显示输入张量

EN

class TensorBoardImage(keras.callbacks.Callback):
def __init__(self, tag):
    super().__init__() 
    self.tag = tag

def on_epoch_end(self, epoch, logs={}):
    # Load image
    img = data.astronaut()
    # Do something to the image
    img = (255 * skimage.util.random_noise(img)).astype('uint8')

    image = make_image(img)
    summary = tf.Summary(value=[tf.Summary.Value(tag=self.tag, image=image)])
    writer = tf.summary.FileWriter('./logs')
    writer.add_summary(summary, epoch)
    writer.close()

    return