转自:https://blog.csdn.net/dss_dssssd/article/details/82977170

https://blog.csdn.net/dss_dssssd/article/details/82980222

https://blog.csdn.net/dss_dssssd/article/details/84103834 

 

pytorch自定义网络层以及损失函数

 

pytorch自定义网络层以及损失函数

下面是使用squential来自定义网络层的例子:

pytorch自定义网络层以及损失函数

 

自定义损失函数:

pytorch自定义网络层以及损失函数

 

pytorch自定义网络层以及损失函数