Python APIs for 2D PointSet Layers