Rename visualization to vis

Summary: Importing from pytorch3d.visualization is wordy, so shortened the path to the vis module and updated the relevant imports.

Reviewed By: nikhilaravi

Differential Revision: D24116527

fbshipit-source-id: e0e4da7d48c5afedec07482d7be43362b6822445

docs/notes/visualization.md

@@ -5,7 +5,7 @@ sidebar_label: Plotly Visualization
 
 # Overview
 
-PyTorch3D provides a modular differentiable renderer, but for instances where we want interactive plots or are not concerned with the differentiability of the rendering process, we provide [functions to render meshes and pointclouds in plotly](../../pytorch3d/visualization/plotly_vis.py). These plotly figures allow you to rotate and zoom the rendered images and support plotting batched data as multiple traces in a singular plot or divided into individual subplots.
+PyTorch3D provides a modular differentiable renderer, but for instances where we want interactive plots or are not concerned with the differentiability of the rendering process, we provide [functions to render meshes and pointclouds in plotly](../../pytorch3d/vis/plotly_vis.py). These plotly figures allow you to rotate and zoom the rendered images and support plotting batched data as multiple traces in a singular plot or divided into individual subplots.
 
 
 # Examples

docs/tutorials/render_colored_points.ipynb

@@ -67,7 +67,7 @@
     "\n",
     "# Data structures and functions for rendering\n",
     "from pytorch3d.structures import Pointclouds\n",
-    "from pytorch3d.visualization import AxisArgs, plot_pointclouds\n",
+    "from pytorch3d.vis import AxisArgs, plot_pointclouds\n",
     "from pytorch3d.renderer import (\n",
     "    look_at_view_transform,\n",
     "    FoVOrthographicCameras, \n",

docs/tutorials/render_textured_meshes.ipynb

@@ -93,7 +93,7 @@
     "\n",
     "# Data structures and functions for rendering\n",
     "from pytorch3d.structures import Meshes\n",
-    "from pytorch3d.visualization import AxisArgs, plot_meshes\n",
+    "from pytorch3d.vis import AxisArgs, plot_meshes\n",
     "from pytorch3d.renderer import (\n",
     "    look_at_view_transform,\n",
     "    FoVPerspectiveCameras, \n",