jit_compile function

Compile TorchScript code into a graph

See the TorchScript language reference for documentation on how to write TorchScript code.

jit_compile(source)

Arguments

• source: valid TorchScript source code.

Examples

if (torch_is_installed()) {
comp <- jit_compile("
def fn (x):
  return torch.abs(x)

def foo (x):
  return torch.sum(x)

")

comp$fn(torch_tensor(-1))
comp$foo(torch_randn(10))
}