小编Leo*_*onG的帖子

如何在PyTorch中使用两个输入构建网络

假设我想拥有通用的神经网络架构:

Input1 --> CNNLayer 
                    \
                     ---> FCLayer ---> Output
                    /
Input2 --> FCLayer
Run Code Online (Sandbox Code Playgroud)

输入1是图像数据,输入2是非图像数据。我已经在Tensorflow中实现了这种架构。

我发现的所有pytorch示例都是通过每一层的一个输入。如何定义前向功能来分别处理2个输入,然后将它们组合在中间层?

python machine-learning computer-vision neural-network pytorch

5
推荐指数
1
解决办法
2999
查看次数