MCPcopy Create free account
hub / github.com/bdsaglam/torch-scae / forward

Method forward

torch_scae/object_decoder.py:393–428  ·  view source on GitHub ↗

Args: obj_encoding: Tensor of shape [B, O, D]. part_pose: Tensor of shape [B, M, P] part_presence: Tensor of shape [B, M] or None; if it exists, it indicates which input parts exist. Returns: A bunch of stuff.

(self,
                obj_encoding: torch.Tensor,
                part_pose: torch.Tensor,
                part_presence: torch.Tensor = None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 1

CapsuleLikelihoodClass · 0.85

Tested by

no test coverage detected