move LinearWithRepeat to pytorch3d

Summary: Move this simple layer from the NeRF project into pytorch3d.

Reviewed By: shapovalov

Differential Revision: D34126972

fbshipit-source-id: a9c6d6c3c1b662c1b844ea5d1b982007d4df83e6
This commit is contained in:
Jeremy Reizenstein
2022-02-14 04:51:02 -08:00
committed by Facebook GitHub Bot
parent ef21a6f6aa
commit 2a1de3b610
6 changed files with 75 additions and 8 deletions

View File

@@ -12,16 +12,16 @@ import torch
from common_testing import TestCaseMixin
from pytorch3d.ops import eyes
from pytorch3d.renderer import (
PerspectiveCameras,
AlphaCompositor,
PointsRenderer,
PerspectiveCameras,
PointsRasterizationSettings,
PointsRasterizer,
PointsRenderer,
)
from pytorch3d.renderer.utils import (
TensorProperties,
ndc_to_grid_sample_coords,
ndc_grid_sample,
ndc_to_grid_sample_coords,
)
from pytorch3d.structures import Pointclouds