MCPcopy Create free account
hub / github.com/NVIDIA/MinkowskiEngine / get_batch_indices

Method get_batch_indices

examples/vae.py:436–437  ·  view source on GitHub ↗
(self, out)

Source from the content-addressed store, hash-verified

434 self.pruning = ME.MinkowskiPruning()
435
436 def get_batch_indices(self, out):
437 return out.coords_man.get_row_indices_per_batch(out.coords_key)
438
439 @torch.no_grad()
440 def get_target(self, out, target_key, kernel_size=1):

Callers 1

sparseMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected