reduce_mean不理解

来源:5-5 Train部分代码编写

关plus

2021-04-16


相关截图:

http://img.mukewang.com/szimg/6078f7e208129fee06600496.jpg老师这里最后卷积层和flatten层中间的reduce_mean是为了把hw降维吗,为什么要这么做呢?是为了减少计算吗?我把这一步去掉发现也能正常训练,网上也没找到详细的解释,可以请您讲解一下吗?

写回答

1回答

会写代码的好厨师

2021-04-20

降低纬度是一方面,另外,可以确保fc层和图片尺寸无关,这样可以做变长的输入。

0
1
关plus
非常感谢!
2021-04-22
共1条回复

Python3+TensorFlow打造人脸识别智能小程序

理论与实战项目双管齐下,让AI技术真正落地应用,适合毕设展示。

1086 学习 · 538 问题

查看课程