Bugfix in reduction feature aggr.

Summary: Bugfix

Reviewed By: bottler

Differential Revision: D40146840

fbshipit-source-id: a8415c361ed3cd939999b87311aff4d3bb1bcfe1

pytorch3d/implicitron/models/view_pooler/feature_aggregator.py

@@ -229,7 +229,7 @@ class ReductionFeatureAggregator(torch.nn.Module, FeatureAggregatorBase):
             masks_sampled.device,
             self.exclude_target_view,
         )
-        aggr_weigths = masks_sampled * sampling_mask
+        aggr_weigths = masks_sampled[..., 0] * sampling_mask[..., None]
         feats_aggregated = {
             k: _avgmaxstd_reduction_function(
                 f,