lab.tensorflow.custom module

lab.tensorflow.custom.as_tf[source]

Convert object to TensorFlow.

Parameters

x (object) – Object to convert.

Returns

x as a TensorFlow object.

Return type

object

lab.tensorflow.custom.tensorflow_register(f, s_f)[source]

Register a function and its sensitivity for TensorFlow.

Parameters

  • f (function) – Function to register.

  • s_f (function) – Sensitivity of f.

Returns

TensorFlow primitive.

Return type

function