sin_op
Structs
Struct: Sin
Fields
Methods
fwd(arg0: Array) -> Array
Computes the sine of the input array element-wise.
jvp(primals: List[Array], tangents: List[Array]) -> Array
Computes the Jacobian-vector product for the sine function.
Args
-
primals
:List[Array]
A list containing the primal input array. -
tangents
:List[Array]
A list containing the tangent vector.
Returns
Array
- The Jacobian-vector product for the sine function.
Implements forward-mode automatic differentiation for the sine function.
Note: The Jacobian-vector product for sine is computed as cos(x) * dx, where x is the primal input and dx is the tangent vector.
vjp(primals: List[Array], grad: Array, out: Array) -> List[Array]
Computes the vector-Jacobian product for the sine function.
Args
-
primals
:List[Array]
A list containing the primal input array. -
grad
:Array
The gradient of the output with respect to some scalar function. -
out
:Array
The output of the forward pass (unused in this function).
Returns
List[Array]
- A list containing the gradient with respect to the input.
Implements reverse-mode automatic differentiation for the sine function.
Note: The vector-Jacobian product for sine is computed as cos(x) * grad, where x is the primal input and grad is the incoming gradient.
unary_simd_op(arg0_real: SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)], arg0_imag: SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]) -> Tuple[SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)], SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]]
Low-level function to compute the sine of a complex number represented as SIMD vectors.
Args
-
arg0_real
:SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]
The real part of the complex number. -
arg0_imag
:SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]
The imaginary part of the complex number.
Returns
Tuple[SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)], SIMD[float32, nelts[::DType]().__mul__(2).__floordiv__(2)]]
- The real and imaginary parts of the sine of the complex number as a tuple.
__call__(mut curr: Array, args: List[Array])
Performs the forward pass for element-wise sine computation of an array.
Args
-
curr
:Array
The current array to store the result (modified in-place). -
args
:List[Array]
A list containing the input array.
Computes the sine of each element in the input array and stores the result in the current array. Initializes the current array if not already set up.
Note: This function assumes that the shape and data of the args are already set up. If the current array (curr) is not initialized, it computes the shape based on the input array and sets up the data accordingly.
Functions
sin
sin(arg0: Array) -> Array