Method squeeze

tensorrt_llm/runtime/session.py:76–80

(self, dim=0)

        return self.shape[0]

    def squeeze(self, dim=0):
        if self.shape[dim] != 1:
            raise ValueError(f"dim {dim} is {self.shape[dim]} instead of 1!")
        return TensorInfo(self.name, self.dtype,
                          self.shape[:dim] + self.shape[dim + 1:])


class Session(object):