D2L-29-Computation-自定义层

Last updated Feb 19, 2022 Edit Source

• All
• Deep learning
• Computation
• Py torch

Tags: #DeepLearning #Computation #PyTorch

• 在线代码实例: 5.4. 自定义层

• 我们可以通过基本层类设计自定义层。这允许我们定义灵活的新层。
• 在自定义层定义完成后，我们就可以在任意环境和网络架构中调用该自定义层。
• 层可以有局部参数，这些参数可以通过内置函数创建。