Skip to Content
DocumentationFunctionalSpacial Opsflip_op

flip_op

View the code on GitHub 

Structs

Struct: Flip

Fields

Methods

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

  • curr: Array

  • args: List[Array]

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

  • primals: List[Array]

  • tangents: List[Array]

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

  • primals: List[Array]

  • grad: Array

  • out: Array

Returns
  • List[Array]
fwd(arg0: Array, dims: List[Int]) -> Array
more details
Args

  • arg0: Array

  • dims: List[Int]

Returns
  • Array

Functions

flip

flip(arg0: Array, dims: List[Int] = List(-1)) -> Array
more details
Args

  • arg0: Array

  • dims: List[Int] (default: List(-1))

Returns
  • Array
Last updated on
conv_transpose3dmax_pool1d_op