docs: def initial_state(self) -> StateTensor:
          '''Returns the initial state tensor.'''
          s0 = []
          for fluent in self._compiler.compile_initial_state(self._batch_size):
              s0.append(self._output_size(fluent))
          s0 = tuple(s0)
          return s0