Save the date! Google I/O returns May 18-20 Register now

tfg.geometry.representation.ray.sample_stratified_1d

Sample points on a ray using stratified sampling.

ray_org A tensor of shape [A1, ..., An, 3], where the last dimension represents the 3D position of the ray origin.
ray_dir A tensor of shape [A1, ..., An, 3], where the last dimension represents the 3D direction of the ray.
near The smallest distance from the ray origin that a sample can have.
far The largest distance from the ray origin that a sample can have.
n_samples A number M to sample on the ray.
name A name for this op that defaults to "stratified_sampling".

A tensor of shape [A1, ..., An, M, 3] indicating the M points on the ray and a tensor of shape [A1, ..., An, M] for the Z values on the points.