Struct Gaussian3dRenderOutputAutodiff

pub struct Gaussian3dRenderOutputAutodiff<AB>
where
    AB: AutodiffBackend,
{
    pub colors_rgb_2d: Tensor<AB, 3>,
    pub positions_2d_grad_norm_ref: Tensor<AB, 1>,
    pub radii: Tensor<<AB as AutodiffBackend>::InnerBackend, 1, Int>,
}

3DGS rendering output (autodiff enabled).

Fields

colors_rgb_2d: Tensor<AB, 3>

2D Colors in RGB space.

The shape is [I_y, I_x, 3].

• I_y: Image height.
• I_x: Image width.

It is the rendered image.

positions_2d_grad_norm_ref: Tensor<AB, 1>

Its gradient is the gradient norm of the 2D positions.

The gradient shape is [P].

• P: Point count.

Usage

use burn::backend::autodiff::grads::Gradients;

let mut grads: Gradients = todo!();

let positions_2d_grad_norm =
    positions_2d_grad_norm_ref.grad_remove(&mut grads);

radii: Tensor<<AB as AutodiffBackend>::InnerBackend, 1, Int>

Visible radii of 3D Gaussians.

The shape is [P].

• P: Point count.

Trait Implementations

impl<AB> Clone for Gaussian3dRenderOutputAutodiff<AB>

where AB: Clone + AutodiffBackend, <AB as AutodiffBackend>::InnerBackend: Clone,

fn clone(&self) -> Gaussian3dRenderOutputAutodiff<AB>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<AB> Debug for Gaussian3dRenderOutputAutodiff<AB>

where AB: AutodiffBackend,

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.