real_op

Structs

Struct: Real

Fields

Methods


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

Computes the shape of 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 compute the real part of.


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

Performs the forward pass for the 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 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.

Note: This function is non-differentiable.

Functions

real


real(arg0: Array) -> Array

Computes the real part of the input array element-wise.

Args

arg0: Array The input array.

Returns

Array - The real part of the input array.