MCPcopy Create free account
hub / github.com/microsoft/TRELLIS / __init__

Method __init__

trellis/modules/sparse/spatial.py:18–20  ·  view source on GitHub ↗
(self, factor: Union[int, Tuple[int, ...], List[int]])

Source from the content-addressed store, hash-verified

16 Implemented as average pooling.
17 """
18 def __init__(self, factor: Union[int, Tuple[int, ...], List[int]]):
19 super(SparseDownsample, self).__init__()
20 self.factor = tuple(factor) if isinstance(factor, (list, tuple)) else factor
21
22 def forward(self, input: SparseTensor) -> SparseTensor:
23 DIM = input.coords.shape[-1] - 1

Callers

nothing calls this directly

Calls 1

__init__Method · 0.45

Tested by

no test coverage detected