( self, x: torch.Tensor, # batch_size * n_tokens * n_channels )
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected