38e284c453bcdad445b300c832ec543f5a0e691e,tensorboard/summary.py,,image,#Any#Any#,149

Before Change

        // try conversion, if failed then need handle by user.
        tensor = np.ndarray(tensor, dtype=np.float32)
    shape = tensor.shape
    height, width, channel = shape[0], shape[1], shape[2]
    if channel == 1:
        // walk around. PIL"s setting on dimension.
        tensor = np.reshape(tensor, (height, width))

After Change

        assert tensor.dim()<4 and tensor.dim()>1, "input tensor should be 3 dimensional."
        if tensor.dim()==2:
            tensor = tensor.unsqueeze(0)
        tensor = tensor.cpu().permute(1,2,0).numpy()
    else:
        tensor = tensor.astype(np.float32)
    tensor = (tensor*255).astype(np.uint8)
    image = make_image(tensor)
    return Summary(value=[Summary.Value(tag=tag, image=image)])

Instances