Method concatenate

flowdock/utils/frame_utils.py:27–32

Concatenate Rigid Transform.

(self, other, dim=0)

        return RigidTransform(self.t.squeeze(dim), self.R.squeeze(dim))

    def concatenate(self, other, dim=0):
        """Concatenate Rigid Transform."""
        return RigidTransform(
            torch.cat([self.t, other.t], dim=dim),
            torch.cat([self.R, other.R], dim=dim),
        )


def get_frame_matrix(