如何获取未知PyTorch模型的输入张量形状

如何获取未知PyTorch模型的输入张量形状