tf_function {tensorflow} | R Documentation |
Creates a callable TensorFlow graph from an R function.
Description
tf_function
constructs a callable that executes a TensorFlow graph created
by tracing the TensorFlow operations in f
. This allows the TensorFlow
runtime to apply optimizations and exploit parallelism in the computation
defined by f
.
Usage
tf_function(f, input_signature = NULL, autograph = TRUE, ...)
Arguments
f |
the function to be compiled |
input_signature |
A possibly nested sequence of |
autograph |
TRUE or FALSE. If TRUE (the default), you can use tensors in
R control flow expressions |
... |
additional arguments passed on to |
Details
A guide to getting started with
tf.function
can
be found here.