Python tensor_list_gather Exemples

Exemple #1

Fichier : list_ops.py Projet : datanonymous/TFandroid

def tensor_list_gather(input_handle, indices, element_dtype, name=None):
  return gen_list_ops.tensor_list_gather(
      input_handle=input_handle,
      indices=indices,
      element_shape=-1,
      element_dtype=element_dtype,
      name=name)

Exemple #2

Fichier : list_ops.py Projet : GiveMeLuna/tensorflow

def _TensorListScatterGrad(op, dlist):
    t, indices, _ = op.inputs
    return gen_list_ops.tensor_list_gather(dlist,
                                           indices,
                                           element_shape=array_ops.slice(
                                               array_ops.shape(t), [1], [-1]),
                                           element_dtype=t.dtype), None, None

Exemple #3

Fichier : list_ops.py Projet : terrytangyuan/tensorflow

def tensor_list_gather(input_handle, indices, element_dtype, name=None):
  return gen_list_ops.tensor_list_gather(
      input_handle=input_handle,
      indices=indices,
      element_shape=-1,
      element_dtype=element_dtype,
      name=name)

Exemple #4

Fichier : list_ops.py Projet : rmlarsen/tensorflow

def _TensorListScatterGrad(op, dlist):
  t, indices, _ = op.inputs
  return gen_list_ops.tensor_list_gather(
      dlist,
      indices,
      element_shape=array_ops.slice(array_ops.shape(t), [1], [-1]),
      element_dtype=t.dtype), None, None

Exemple #5

Fichier : list_ops.py Projet : Chiranjeevi731/Machine-Learning

def _TensorListScatterIntoExistingListGrad(op, dlist):
  """Gradient function for TensorListScatterIntoExistingList."""
  _, tensor, indices = op.inputs
  dtensor = gen_list_ops.tensor_list_gather(
      dlist,
      indices,
      element_shape=array_ops.slice(array_ops.shape(tensor), [1], [-1]),
      element_dtype=tensor.dtype)
  zeros = array_ops.zeros_like(tensor)
  dlist = tensor_list_scatter(zeros, indices, indices, input_handle=dlist)
  return dlist, dtensor, None

Exemple #6

Fichier : list_ops.py Projet : adit-chandra/tensorflow

def _TensorListScatterIntoExistingListGrad(op, dlist):
  """Gradient function for TensorListScatterIntoExistingList."""
  _, tensor, indices = op.inputs
  dtensor = gen_list_ops.tensor_list_gather(
      dlist,
      indices,
      element_shape=array_ops.slice(array_ops.shape(tensor), [1], [-1]),
      element_dtype=tensor.dtype)
  zeros = array_ops.zeros_like(tensor)
  dlist = tensor_list_scatter(zeros, indices, indices, input_handle=dlist)
  return dlist, dtensor, None

Exemple #7

Fichier : list_ops.py Projet : datanonymous/TFandroid

def _TensorListScatterGrad(op, dlist):
  """Gradient function for TensorListScatter."""
  tensor = op.inputs[0]
  indices = op.inputs[1]
  dtensor = gen_list_ops.tensor_list_gather(
      dlist,
      indices,
      element_shape=array_ops.slice(array_ops.shape(tensor), [1], [-1]),
      element_dtype=tensor.dtype)
  if op.type == "TensorListScatterV2":
    return dtensor, None, None, None
  else:
    return dtensor, None, None

Exemple #8

Fichier : list_ops.py Projet : terrytangyuan/tensorflow

def _TensorListScatterGrad(op, dlist):
  """Gradient function for TensorListScatter."""
  tensor = op.inputs[0]
  indices = op.inputs[1]
  dtensor = gen_list_ops.tensor_list_gather(
      dlist,
      indices,
      element_shape=array_ops.slice(array_ops.shape(tensor), [1], [-1]),
      element_dtype=tensor.dtype)
  if op.type == "TensorListScatterV2":
    return dtensor, None, None, None
  else:
    return dtensor, None, None

Exemple #9

Fichier : list_ops.py Projet : becster/tensorflow

def _TensorListScatterGrad(op, dlist):
  t, indices, _ = op.inputs
  return gen_list_ops.tensor_list_gather(
      dlist, indices, element_dtype=t.dtype), None

Exemple #10

Fichier : list_ops.py Projet : qwerzou1/shibie

def _TensorListScatterGrad(op, dlist):
    t, indices, _ = op.inputs
    return gen_list_ops.tensor_list_gather(dlist,
                                           indices,
                                           element_dtype=t.dtype), None