Package ai.djl.pytorch.engine
Class PtGradientCollector
- java.lang.Object
-
- ai.djl.pytorch.engine.PtGradientCollector
-
- All Implemented Interfaces:
ai.djl.training.GradientCollector,java.lang.AutoCloseable
public class PtGradientCollector extends java.lang.Object implements ai.djl.training.GradientCollectorPtGradientCollectoris the PyTorch implementation ofGradientCollector.
-
-
Constructor Summary
Constructors Constructor Description PtGradientCollector()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description voidbackward(ai.djl.ndarray.NDArray target)voidclose()
-