Is there a Python API to access any of the aten ops?  How could one reproduce a network with specific aten ops? For example, how could I recreate a single operator model that uses the aten::index operator? Is there documentation for the ops somewhere?