view_as_real_op

Structs

Struct: ViewAsReal

Fields

Methods


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

Computes the shape of an array after viewing it as the real 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 real part.


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

Performs the forward pass for the view_as_real 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_real view is created.

Note: The information of the shape computation is stored in the ArrayShape object of the curr 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) -> Array

Creates a view of the input array as a real array.

Args

arg0: Array The input array.

Returns

Array - A view of the input array as a real array.

Functions

view_as_real


view_as_real(arg0: Array) -> Array

Creates a view of the input array as the real part of a complex array.

Args

arg0: Array The input array.

Returns

Array - A view of the input array as the real part of a complex array.