A turbo-machine is provided having at least one turbo-machine stage, wherein overload protection of the rotating shaft is provided by the method and, to protect the rotating shaft against overload, a measurement of a torsional stress of the rotating shaft is carried out during operation of the rotating shaft. The torsional stress is measured via a measurement of the torque of the rotating shaft. The torque is preferably measured magneto-elastically. By using the measured torque of the rotating shaft, the development of the torque is forecast. On the basis of the forecast produced, the turbo-machine stage or operating parameters of the turbo-machine is/are regulated. Further a turbo-machine is provided comprising at least one turbo-machine stage which has at least one rotating shaft, wherein the turbo-machine has a device for carrying out the method.