fix error on set_mask, fix error on extra dim after event selection and fix error and square matrix for deepjdot