title: Converting TensorShape([1, 600, 1200, 3]) to png
body: <p>My model is outputting an image of size: <code>TensorShape([1, 600, 1200, 3])</code></p>
  <p>I'm attempting to output this image like this:</p>
  <pre><code>from PIL import Image
  Image.fromarray(np.asarray(tensor_image)).show()
  </code></pre>
  <p>I get the following error:</p>
  <pre><code>TypeError: Cannot handle this data type: (1, 1, 1200, 3), &lt;f4
  </code></pre>
  <p>I'm not sure how I should resize the tensor.</p>