|
|
Home | Main Page | Topics | Namespace List | Class Hierarchy | Alphabetical List | Data Structures | File List | Namespace Members | Data Fields | Globals | Related Pages |
#include <ImpactLoss.h>
Combined cosine similarity and exponential L1 loss.
Useful for simultaneously penalizing direction and magnitude.
Definition at line 356 of file ImpactLoss.h.
Public Member Functions | |
| L1Cosine () | |
| void | updateValue (torch::Tensor &fixedOutput, torch::Tensor &movingOutput) override |
| torch::Tensor | updateValueAndGetGradientModulator (torch::Tensor &fixedOutput, torch::Tensor &movingOutput) override |
| Public Member Functions inherited from ImpactLoss::Loss | |
| virtual torch::Tensor | GetDerivative (double N) const |
| virtual double | GetValue (double N) const |
| virtual void | initialize (torch::Tensor &output) |
| Loss (bool isLossNormalized) | |
| virtual Loss & | operator+= (const Loss &other) |
| void | reset () |
| void | setNumberOfParameters (int numberOfParameters) |
| void | updateDerivativeInJacobianMode (torch::Tensor &jacobian, torch::Tensor &nonZeroJacobianIndices) |
| virtual void | updateValueAndDerivativeInStaticMode (torch::Tensor &fixedOutput, torch::Tensor &movingOutput, torch::Tensor &jacobian, torch::Tensor &nonZeroJacobianIndices) |
| virtual | ~Loss ()=default |
Private Attributes | |
| double | m_Lambda |
Additional Inherited Members | |
| Protected Attributes inherited from ImpactLoss::Loss | |
| torch::Tensor | m_Derivative |
| bool | m_Initialized = false |
| int | m_NumberOfParameters |
| double | m_Value |
|
inline |
Definition at line 362 of file ImpactLoss.h.
|
inlineoverridevirtual |
Implements ImpactLoss::Loss.
Definition at line 369 of file ImpactLoss.h.
|
inlineoverridevirtual |
Implements ImpactLoss::Loss.
Definition at line 381 of file ImpactLoss.h.
|
private |
Definition at line 359 of file ImpactLoss.h.
Generated on 1774142652 for elastix by 1.15.0 |