pytorch打印每一层的模型结构

  • 2024-10-20