Skip to Content
DocumentationArray and Nodeutils

utils

View the code on GitHub 

Structs

Functions

default_fwd

default_fwd(mut curr: Array, args: List[Array])
more details
Args

  • curr: Array

  • args: List[Array]

default_vjp

default_vjp(primals: List[Array], grad: Array, out: Array) -> List[Array]
more details
Args

  • primals: List[Array]

  • grad: Array

  • out: Array

Returns
  • List[Array]

default_jvp

default_jvp(primals: List[Array], tangents: List[Array]) -> Array
more details
Args

  • primals: List[Array]

  • tangents: List[Array]

Returns
  • Array
Last updated on
nodefunctional