Module custom_ops 

Custom operations with user-defined functions

This module provides support for custom operations where users can define their own reduction functions, element-wise operations, and more.

Functions

apply_custom_unary

Apply a custom operation to a tensor handle

custom_binary_op

Custom element-wise binary operation with user-defined function

custom_reduce

Custom reduction operation with user-defined reduction function

custom_unary_op

Custom element-wise unary operation with user-defined function