T3Tangent.tangent_stack_shape

t3toolbox.manifold.T3Tangent.tangent_stack_shape()

def tangent_stack_shape(self) -> typ.Tuple[int, ...]:

Tangent stack K: the extra outer batch of tangent vectors sharing this frame.

This is the part of the variation stack that exceeds the frame stack (often empty). The variation cores are stacked as K + C + (core,) – extra axes outermost, frame stack inner.

Return type:

t3toolbox.backend.common.typ.Tuple[int, Ellipsis]