(128, 2, 224, 224, 3) image a shape (128, 224, 224, 3) labels tf.Tensor(2, shape=(), dtype=int32) prediction a tf.Tensor([ 0.15290104 -0.11335965 0.05952606 0.63431394 -0.6388165 ], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3) image a shape (128, 224, 224, 3) labels tf.Tensor(0, shape=(), dtype=int32) prediction a tf.Tensor([ 1.2271674 0.06227781 -0.06310828 0.509202 0.16954662], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3) image a shape (128, 224, 224, 3) labels tf.Tensor(4, shape=(), dtype=int32) prediction a tf.Tensor([-0.34474376 0.09270601 -0.44066426 -0.08329549 -0.04816262], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3) image a shape (128, 224, 224, 3) labels tf.Tensor(0, shape=(), dtype=int32) prediction a tf.Tensor([ 0.8306643 0.03662108 -0.40713277 -0.44725904 -0.27027306], shape=(5,), dtype=float32)
(128, 2, 224, 224, 3) image a shape (128, 224, 224, 3) labels tf.Tensor(3, shape=(), dtype=int32) prediction a tf.Tensor([ 0.6583673 -0.3735458 0.00993944 1.0526401 0.26237836], shape=(5,), dtype=float32)