Method split

tensorrt_llm/functional.py:504–508

See functional.split.

(self, split_size_or_sections, dim=0)

        return self.rank()

    def split(self, split_size_or_sections, dim=0):
        '''
        See functional.split.
        '''
        return split(self, split_size_or_sections, dim)

    def select(self, dim, index):
        '''