tfe extract return np array