Minkowski修剪
Minkowski修剪
Class MinkowskiEngine.MinkowskiPruning
從MinkowskiEngine.SparseTensor中刪除指定的坐標(biāo)。
init（）
初始化由nn.Module和ScriptModule共享的內(nèi)部模塊狀態(tài)。
forward(input, mask)
Args：
input（MinkowskiEnigne.SparseTensor）：從中刪除坐標(biāo)的稀疏張量。
mask（torch.BoolTensor）：指定要保留哪一個(gè)的掩碼向量。False的坐標(biāo)將被刪除。
返回值：
A MinkowskiEngine.SparseTensor，C =對(duì)應(yīng)于mask == True的坐標(biāo)F =來(lái)自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

總結(jié)

以上是生活随笔為你收集整理的Minkowski修剪的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。