MinkowskiPruning¶
MinkowskiPruning¶
-
class
MinkowskiEngine.
MinkowskiPruning
¶ Remove specified coordinates from a
MinkowskiEngine.SparseTensor
.-
__init__
()¶ Initializes internal Module state, shared by both nn.Module and ScriptModule.
-
forward
(input: MinkowskiSparseTensor.SparseTensor, mask: torch.Tensor)¶ - Args:
input
(MinkowskiEnigne.SparseTensor
): a sparse tensor to remove coordinates from.mask
(torch.BoolTensor
): mask vector that specifies which one to keep. Coordinates with False will be removed.- Returns:
A
MinkowskiEngine.SparseTensor
with C = coordinates corresponding to mask == True F = copy of the features from mask == True.
Example:
>>> # Define inputs >>> input = SparseTensor(feats, coords=coords) >>> # Any boolean tensor can be used as the filter >>> mask = torch.rand(feats.size(0)) < 0.5 >>> pruning = MinkowskiPruning() >>> output = pruning(input, mask)
-
training
: bool¶
-