Ejemplos de sparse_segment_mean_grad en Python

Ejemplo n.º 1

Archivo: math_grad.py Proyecto: Hwhitetooth/tensorflow

def _SparseSegmentMeanGrad(op, grad):
  """Gradient for SparseSegmentMean."""
  dim0 = array_ops.shape(op.inputs[0])[0]
  return (math_ops.sparse_segment_mean_grad(grad,
                                            op.inputs[1],
                                            op.inputs[2],
                                            dim0),
          None, None)

Ejemplo n.º 2

Archivo: math_grad.py Proyecto: rsandstroem/tensorflow

def _SparseSegmentMeanGrad(op, grad):
  """Gradient for SparseSegmentMean."""
  dim0 = array_ops.shape(op.inputs[0])[0]
  return (math_ops.sparse_segment_mean_grad(grad,
                                            op.inputs[1],
                                            op.inputs[2],
                                            dim0),
          None, None)