Tensorflow custom layer accessing value of output tensor from previous layer
So I’d like to create a custom Tensorflow.Keras layer with purpose of returning label name as a text.
The previous layer is an argmax that returns integer representing the index of predicted class and I want to pass it as input to the next layer which returns the real name of the class as the final output of the model.
dataset of tensors shuffled after applying map or model.predict
I am trying to predict a target image from an input image and then display both the input and prediction. For some reason, test_images
seems to be reshuffled in the process so I can’t get them to line up. test_dataset
contains a tuple with two tensors, one of the input images and the other of the target images.