gdhy9064

You can try to use `np.array(self.image_data)` instead of `jt.array(self.image_data)`, and change ```python if self.img_alpha and self.image_data.shape[-1]==3: self.image_data=jt.concat([self.image_data,jt.ones(self.image_data.shape[:-1]+(1,))],-1).stop_grad() ``` to ```python if self.img_alpha and self.image_data.shape[-1]==3: self.image_data=np.concatenate([self.image_data,np.ones([*self.image_data.shape[:-1],1])],-1) ``` as well as ```python...