view_as_imag_op

Structs

Struct: ViewAsImag

Fields

Methods


compute_shape(mut curr: ArrayShape, args: List[ArrayShape])

Computes the shape of an array after viewing it as the imaginary part of a complex array.

Args

curr: ArrayShape The ArrayShape to store the result of the computation.
args: List[ArrayShape] The ArrayShape to view as the imaginary part.


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

Performs the forward pass for the view_as_imag operation. It sets the base of the argument to be the base of the current array and computes the shape of the current array via its dedicated ArraySahpe fwd fucntion.

Args

curr: Array The current array to store the result (modified in-place).
args: List[Array] The array on which the view_as_imag view is created.

Note: The information of the shape computation is stored in the ArrayShape object of the curr array.


fwd(arg0: Array) -> Array

Creates a view of the input array as an imaginary array.

Args

arg0: Array The input array.

Returns

Array - A view of the input array as an imaginary array.

Note: This function is non-differentiable.


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]

Functions

view_as_imag


view_as_imag(arg0: Array) -> Array

Creates a view of the input array as an imaginary array.

Args

arg0: Array The input array.

Returns

Array - A view of the input array as an imaginary array.

Note: This function is non-differentiable.