Operators for neural networks¶
add_loss(loss: oneflow._oneflow_internal.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.
loss – A Blob.