Several simple examples showing the usage of scatter_nd_sub

最新推荐文章于 2024-09-04 07:30:00 发布

sclxf

最新推荐文章于 2024-09-04 07:30:00 发布

阅读量511

点赞数

分类专栏： Tensorflow

本文链接：https://blog.csdn.net/sclxf/article/details/80238724

版权

Tensorflow 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

scatter_nd_sub

Example 1: subtraction at the second dimension, [0,1], which corresponding to a vector.

>>> ref = tf.Variable(tf.ones([2,3,3],tf.int32))        
>>> indices = tf.constant([[0,1]])
>>> updates = tf.constant([[1,1,1]])    
>>> init = tf.global_variables_initializer()
>>> sess.run(init)
>>> print(ref.eval())
[[[1 1 1]
  [1 1 1]
  [1 1 1]]

 [[1 1 1]
  [1 1 1]
  [1 1 1]]]
>>> print(updates.eval())
[[1 1 1]]
>>> update = tf.scatter_nd_sub(ref,indices,updates)
>>> sess.run(update)
array([[[1, 1, 1],
    [0, 0, 0],
    [1, 1, 1]],

   [[1, 1, 1],
    [1, 1, 1],
    [1, 1, 1]]], dtype=int32)

Example 2: subtraction at the third dimension, [0,1,2], which points to a single element.

>>> ref = tf.Variable(tf.ones([2,3,3],tf.int32))
>>> indices = tf.constant([[0,1,2]])
>>> updates = tf.constant([1])
>>> init = tf.global_variables_initializer()
>>> sess.run(init)
>>> print(ref.eval())
[[[1 1 1]
  [1 1 1]
  [1 1 1]]

 [[1 1 1]
  [1 1 1]
  [1 1 1]]]
>>> print(updates.eval())
[1]
>>> update = tf.scatter_nd_sub(ref,indices,updates)
>>> sess.run(update)
array([[[1, 1, 1],
    [1, 1, 0],
    [1, 1, 1]],

   [[1, 1, 1],
    [1, 1, 1],
    [1, 1, 1]]], dtype=int32)