Method rank

tensorrt_llm/functional.py:492–496

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

(self)

        return self.trt_tensor.shape[dim]

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

    def ndim(self):
        '''