Method numel

tensorrt_llm/runtime/session.py:53–54

(self)

    # add more info like strides, formats if needed

    def numel(self):
        return prod(self.shape)

    def view(self, *shape):
        assert set(map(type, shape)) == {int}