Exports
-
newDiffs, no_grad_mode, Gate, check_ctx, newParents, register_node, PayloadKind, backprop, is_grad_needed, Context, newContext, Backward, Payload, Variable, SmallDiffs, variable, -, AddGate, SubGate, +, MatMulGate, *, SumGate, MeanGate, sum, mean, mean, squeeze, flatten, reshape, unsqueeze, reshape, ReshapeGate, [], ChunkSplitGate, stack, chunk, *., HadamardGate