Minkowski修剪

Minkowski修剪

Minkowski修剪

Class MinkowskiEngine.MinkowskiPruning

从MinkowskiEngine.SparseTensor中删除指定的坐标。

__init__（）

初始化由nn.Module和ScriptModule共享的内部模块状态。

forward(input, mask)

Args：

input（MinkowskiEnigne.SparseTensor）：从中删除坐标的稀疏张量。

mask（torch.BoolTensor）：指定要保留哪一个的掩码向量。False的坐标将被删除。

返回值：

A MinkowskiEngine.SparseTensor，C =对应于mask == True的坐标F =来自mask == True的特征的副本。

例：

>>> # 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