class_docstr: A wrapper that turns a callable into a valid Model
class_docstr_tok: ["A","wrapper","that","turns","a","callable","into","a","valid","Model"]
docstr: Wrap a callable function that takes a tensor as input and returns
  a tensor as output with the given layer name.
  :param callable_fn: The callable function taking a tensor and
                      returning a given layer as output.
  :param output_layer: A string of the output layer returned by the
                       function. (Usually either "probs" or "logits".)
docstr_tok: ["Wrap","a","callable","function","that","takes","a","tensor","as","input","and","returns","a","tensor","as","output","with","the","given","layer","name",".",":","param","callable_fn",":","The","callable","function","taking","a","tensor","and","returning","a","given","layer","as","output",".",":","param","output_layer",":","A","string","of","the","output","layer","returned","by","the","function",".","(","Usually","either","``","probs","''","or","``","logits","''",".",")"]
code: def __init__(self, callable_fn, output_layer):
      """
      Wrap a callable function that takes a tensor as input and returns
      a tensor as output with the given layer name.
      :param callable_fn: The callable function taking a tensor and
                          returning a given layer as output.
      :param output_layer: A string of the output layer returned by the
                           function. (Usually either "probs" or "logits".)
      """
  
      super(CallableModelWrapper, self).__init__()
      self.output_layer = output_layer
      self.callable_fn = callable_fn
  
imports: from abc import ABCMeta
  import warnings
  import tensorflow
  from cleverhans import utils_tf
label_desc: Use image node_id 1 to create a new CallableModelWrapper object from inherited base classes: Model with example: obj = CallableModelWrapper(callable_fn, output_layer)