Dynadot Jtti 搬瓦工 腾讯云
共 1 篇文章

标签:一个网络多个输出

Pytorch 同一个网络同时拥有多个输出的写法-老唐笔记

Pytorch 同一个网络同时拥有多个输出的写法

Pytorch 是非常热门的神经网络计算框架,一般来说一个网络对应一个输出,如果我们需要一个网络同时输出多个 output,这在 Pytorch 也是可以做到的。输出之后,可以分别计算 Loss,然后进行 BP。 一、同一个网络输出多个结果...

赞(6)老唐老唐科研