primalAndJvp

fun primalAndJvp(x: DTensor, v: DTensor, f: (DTensor) -> DTensor): Pair<DTensor, DTensor>

Content copied to clipboard

fun <Input : Any, Output : Any, InputTangent : Any, OutputTangent : Any> primalAndJvp(x: Input, v: InputTangent, f: (Input) -> Output, makeForwardInput: (Input, InputTangent, makeForwardTensor: (primal: DTensor, tangent: DTensor) -> DTensor) -> Input, wrapOutput: (Output, Wrapper) -> Output? = null, extractTangent: (Output, extractTensorTangent: (outputTensor: DTensor) -> DTensor) -> OutputTangent): Pair<Output, OutputTangent>

Content copied to clipboard

fun <Input : Any, Output : Any, OutputTangent : Any> primalAndJvp(x: Input, v: Input, f: (Input) -> Output, wrapInput: (Input, Wrapper) -> Input? = null, wrapOutput: (Output, Wrapper) -> Output? = null, extractOutputTangent: (Output, extractTensorTangent: (outputTensor: DTensor) -> DTensor) -> OutputTangent): Pair<Output, OutputTangent>

Content copied to clipboard

fun <Input : Any, Output : Any> primalAndJvp(x: Input, v: Input, f: (Input) -> Output, wrapInput: (Input, Wrapper) -> Input? = null, wrapOutput: (Output, Wrapper) -> Output? = null): Pair<Output, Output>

Content copied to clipboard