tensorflow/sig-jvm

Hello, we run Tensorflow Java 0.2.0 (TF 2.3.1). And we have a model that produces an image out of the model as a byte[]. This is fine when running the model in python. We can write the bytes to a file. But when trying to do the same thing using TF Java we run into getting too few bytes from the output. I think I have managed to boil it down to unit test with TString.

    public void testTFNdArray() throws Exception {
        ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
        byte[] readAllBytes = Files.readAllBytes(Path.of(contextClassLoader.getResource("img_12.png").getFile()));
        NdArray<byte[]> vectorOfObjects = NdArrays.vectorOfObjects(readAllBytes);

        Tensor<TString> tensorOfBytes = TString.tensorOfBytes(vectorOfObjects);
        TString data = tensorOfBytes.data();

        byte[] asBytes = tensorOfBytes.data().asBytes().getObject(0);

        System.out.println("Bytes original file: " + readAllBytes.length);
        System.out.println("NdArray byte[] length: " + vectorOfObjects.getObject().length);
        System.out.println("Tensor numbytes: " + tensorOfBytes.numBytes());
        System.out.println("TString size: " + data.size());
        System.out.println("Bytes with reading from TString (WRONG):  " + asBytes.length);
    }

This is the problem that I get with running through a real model. How should we be able to get the full byte[] out again?