如何减少原型网络代码

Question

我有如下代码。 输入矩阵为（90，2856）。

我想将此输入用于 PrototypicalNet（nn.Module）类：不包含Net（nn.Module）类：

请让我知道如何删除类Net（nn.Module）的部分：

这是原型网络的完整代码。

我想删除类Net（nn.Module）的一部分，因为我已经使用了其他降维方法。

Few-Shot Learning With Prototypical Networks