Method ndim

tensorrt_llm/functional.py:498–502

Returns the rank (i.e. the number of dimensions) of the tensor.

(self)

        return len(self.trt_tensor.shape)

    def ndim(self):
        '''
        Returns the rank (i.e. the number of dimensions) of the tensor.
        '''
        return self.rank()

    def split(self, split_size_or_sections, dim=0):
        '''