oneflow.losses¶
Operators for neural networks¶
-
oneflow.losses.
add_loss
(loss: oneflow_api.BlobDesc) → None¶ Mark a Blob as a loss. Auto grad starts at every loss blob. It doesn’t has to be a product of typical “loss” operator like softmax loss but can also be a Blob produced by any operator.
- Parameters
loss – A Blob.