自动微分 -凯发k8网页登录

自定义深度学习层、网络、训练循环和损失函数

对于大多数任务，您可以使用内置层。如果没有您的任务所需的内置层，则可以定义您自己的自定义层。您可以使用自定义输出层指定自定义损失函数，并定义具有可学习参数和状态参数的自定义层。定义自定义层后，您可以检查该层是否有效，是否与 gpu 兼容，以及是否输出正确定义的梯度。要查看支持的层的列表，请参阅。

如果函数不提供任务所需的训练选项，或者自定义输出层不支持所需的损失函数，则您可以定义自定义训练循环。对于层图不支持的模型，您可以将自定义模型定义为函数。要了解详细信息，请参阅。