code: def loss(x, label):
      return objax.functional.loss.sigmoid_cross_entropy_logits(
          model(x)[:, 0], label
      ).mean()
  
file_path: objax/examples/image_classification/horses_or_humans_logistic.py
imports: import os
  import numpy
  import tensorflow_datasets
  import objax
  from objax.util import EasyDict
label_desc: Use image node_id 2 for calling a global function with example usage: loss(x, label) and returns: objax