python将tensor写入npz