Aten op documentation and Python API

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?