res/TensorFlowPythonExamples/examples/segment_sum/__init__.py

   1 import tensorflow as tf
   2
   3 lhs_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(4, 4, 4, 4), name="Hole")
   4 rhs_ = tf.compat.v1.placeholder(dtype=tf.int32, shape=(4, ), name="Hole")
   5 op_ = tf.compat.v1.math.segment_sum(lhs_, rhs_)