Can you share the error details as where the error is experienced in the execution ?
the error means your tensor is not big enough for the shape you are trying to reshape it to.
(1,3,64,64) should be create from a 1D tensor of shape (12288)