See the TorchScript language reference for documentation on how to write TorchScript 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))
}
#> torch_tensor
#> -2.62729
#> [ CPUFloatType{} ]