I think it makes sense to just save the tensor as image without modifying it. But looks like it modifies it.