tensorplay.autograd.function

Classes

class Function [source]

Function()

Records operation history and defines formulas for differentiating ops.

Methods

apply(*args, **kwargs) [source]

---

backward(ctx, *grad_outputs) [source]

Defines a formula for differentiating the operation.

---

forward(ctx, *args, **kwargs) [source]

Performs the operation.

---