About the ops_index in 'new_model.py'

[MJITG]

Thanks for your great jobs again! I'm confused about the 'self.ops_index' in 'new_model.py'. I guess it should be fetched in the 'self.cell_arch' (which is a tensor containing branch index and PRIMITIVES index in pairs) according to the branch index instead of increasing every time. If not, the input and the operation of one block may be mismatched.