PyTorch像Keras模型一样获得输入和输出

Question

如何轻松获取PyTorch模型的输入和输出？使用Keras，我可以做到：

model_inputs = model.inputs
model_output = model.layers[-1].output

我的PyTorch模型没有任何类似的功能。