Relatively new to tensorflow here, and I am facing an issue where I have not yet managed to find a good answer through searching. So here goes: