Update rasterize docstring to correct differentiability behavior.

PiperOrigin-RevId: 673387775
google-research · Sep 11, 2024 · 0cd0d4d · 0cd0d4d
1 parent 7c660b3
commit 0cd0d4d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/diffren/jax/internal/rasterize.py b/diffren/jax/internal/rasterize.py
@@ -36,7 +36,8 @@ def rasterize_triangles(vertices: jnp.ndarray,
   If num_layers > 1, the layers are returned in front-to-back order (layer 0 is
   closest to the camera).
 
-  None of the outputs of this function are differentiable.
+  None of the outputs of this function are differentiable unless
+  compute_diff_barys is True.
 
   Args:
     vertices: float array of xyz positions with shape [vertex_count, d]. If