* [PT FE]: support aten::index * bool indexing testing * more tests, fix nonzero case * apply code review