当前位置: 首页>>代码示例>>Python>>正文


Python gradients_impl._AggregateIndexedSlicesGradients函数代码示例

本文整理汇总了Python中tensorflow.python.ops.gradients_impl._AggregateIndexedSlicesGradients函数的典型用法代码示例。如果您正苦于以下问题:Python _AggregateIndexedSlicesGradients函数的具体用法?Python _AggregateIndexedSlicesGradients怎么用?Python _AggregateIndexedSlicesGradients使用的例子?那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。


在下文中一共展示了_AggregateIndexedSlicesGradients函数的5个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: testMixedTensorAndIndexedSlices

 def testMixedTensorAndIndexedSlices(self):
   t0 = math_ops._as_indexed_slices(constant_op.constant(
       [[1., 2.], [0, 0], [3., 4.]]))
   t1 = constant_op.constant(
       [[0., 0.], [5, 6], [7., 8.]])
   total = constant_op.constant(
       [[1., 2.], [5, 6], [10., 12.]])
   result = gradients_impl._AggregateIndexedSlicesGradients([t0, t1])
   self._assert_indexed_slices_equal(total, result)
开发者ID:jinxin0924,项目名称:tensorflow,代码行数:9,代码来源:gradients_test.py

示例2: testMultipleGradientsWithNones

 def testMultipleGradientsWithNones(self):
   t0 = math_ops._as_indexed_slices(constant_op.constant(
       [[1., 2.], [0, 0], [3., 4.]]))
   t1 = math_ops._as_indexed_slices(constant_op.constant(
       [[0., 0.], [5, 6], [7., 8.]]))
   t3 = None
   total = constant_op.constant(
       [[1., 2.], [5, 6], [10., 12.]])
   result = gradients_impl._AggregateIndexedSlicesGradients([t0, t1, t3])
   self._assert_indexed_slices_equal(total, result)
开发者ID:jinxin0924,项目名称:tensorflow,代码行数:10,代码来源:gradients_test.py

示例3: aggregate_tensors_or_indexed_slices

def aggregate_tensors_or_indexed_slices(values, accumulation_fn=math_ops.add_n):
  """Aggregate tensors using `accumulation_fn` and IndexedSlices via concat."""
  if any(isinstance(v, ops.IndexedSlices) for v in values):
    return gradients_impl._AggregateIndexedSlicesGradients(values)  # pylint: disable=protected-access
  else:
    return accumulation_fn(values)
开发者ID:abhinav-upadhyay,项目名称:tensorflow,代码行数:6,代码来源:cross_tower_utils.py

示例4: testOneGradient

 def testOneGradient(self):
   t = math_ops._as_indexed_slices(constant_op.constant(
       [[1., 2.], [0, 0], [3., 4.]]))
   result = gradients_impl._AggregateIndexedSlicesGradients([t])
   self._assert_indexed_slices_equal(t, result)
开发者ID:jinxin0924,项目名称:tensorflow,代码行数:5,代码来源:gradients_test.py

示例5: testNoGradients

 def testNoGradients(self):
   self.assertIsNone(gradients_impl._AggregateIndexedSlicesGradients([]))
开发者ID:jinxin0924,项目名称:tensorflow,代码行数:2,代码来源:gradients_test.py


注:本文中的tensorflow.python.ops.gradients_impl._AggregateIndexedSlicesGradients函数示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。