torch.register_function
function register_function<Name extends string, F extends (...args: any[]) => any>(config: FunctionConfig<Name, F>): FunctionHandle<Name>function register_function<Name extends string, F extends (...args: any[]) => any>(name: Name, impl: F, options?: RegisterFunctionOptions): FunctionHandle<Name>Register a custom function with full configuration object.
Parameters
configFunctionConfig<Name, F>
Returns
FunctionHandle<Name>Examples
torch.library.register_function({
name: 'quaternion_mul',
namespace: 'quaternion',
impl: (a, b) => {
// quaternion multiplication
},
});
// Usage: torch.quaternion.quaternion_mul(q1, q2)