torch.cuda.jiterator._create_jit_fn
torch.cuda.jiterator._create_jit_fn(code_string, **kwargs)
Create a jiterator-generated cuda kernel for an elementwise op.
The code string has to be a valid CUDA function that describes the
   computation for a single element. The code string has to follow the
   c++ template pattern, as shown in the example below. This function
   will be inlined into elementwise kernel template, and compiled on
   the fly. Compiled kernel will be cached in memory, as well as local
   temp dir.
Jiterator-generated kernels accepts noncontiguous tensors, and
   supports boardcasting and type promotion.
Parameters:
      * code_string (str) -- CUDA code string to be compiled by
        jiterator. The entry functor must return by value.
  * **kwargs** (*Dict**, **optional*) -- Keyword arguments for
    generated function

Return type:
      Callable
Example: