torch.RegisterFunctionOptions
export interface RegisterFunctionOptions {
/**
* Optional namespace for the function.
* If specified, function will be at torch[namespace].funcName.
*/
readonly namespace?: string;
/**
* Optional signature description for documentation.
*/
readonly signature?: string;
/**
* Optional dtype restriction.
* If specified, function only works with these dtypes.
*/
readonly dtypes?: readonly DType[];
}- readonly
namespace(string)optional - – Optional namespace for the function. If specified, function will be at torch[namespace].funcName.
- readonly
signature(string)optional - – Optional signature description for documentation.
- readonly
dtypes(readonly DType[])optional - – Optional dtype restriction. If specified, function only works with these dtypes.
Options for register_function.