Torchscript/C++ jit：：跟踪模型-访问层参数

Question

我有一个用python训练的模型，使用torch.jit.trace进行跟踪，并使用torch::jit::load加载到C++。

是否有方法访问最后一层以提取模型所需输出深度的值(例如，如果是从16 Conv2D 2开始的->层，我想为深度b，d->2，x，y=2预先定义一个张量)？

Answer 1

这不是解决这个问题的最优雅的方法，但最简单的方法就是通过一个虚拟张量并访问这个形状。我尝试的另一种方法是访问参数列表并查找"softmax"，不幸的是，我不能保证每个人的模型在搜索时都会以相同的方式拼写它。如果其他人对此有一个很好的答案，请随意分享，但这将是目前必须做的。