其他分享
首页 > 其他分享> > Minkowski修剪

Minkowski修剪

作者:互联网

Minkowski修剪

Minkowski修剪

Class MinkowskiEngine.MinkowskiPruning

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

__init__()

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

forward(input, mask)

Args

inputMinkowskiEnigne.SparseTensor):从中删除坐标的稀疏张量。

masktorch.BoolTensor):指定要保留哪一个的掩码向量。False的坐标将被删除。

返回值:

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)

trainingbool

 

标签:修剪,MinkowskiEngine,mask,坐标,Minkowski,input,SparseTensor
来源: https://www.cnblogs.com/wujianming-110117/p/14227711.html