code: def _searchsorted(sorted_sequence, values):
      """
      Searches for which bin an input belongs to (in a way that is parallelizable and
      amenable to autodiff)
  
      TODO: Replace with torch.searchsorted once it is released
      """
      return torch.sum(values[..., None] >= sorted_sequence, dim=-1) - 1
  
imports: from functools import partial
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.distributions import Transform, constraints
  from pyro.nn import DenseNN
  from ..conditional import ConditionalTransformModule
  from ..torch_transform import TransformModule
  from ..util import copy_docs_from
label_desc: Use image node_id 1 for calling a global function with example usage: _searchsorted(sorted_sequence, values) and returns: unknown