2017-03-20 2 views
1

나는 같은 랭크의 텐서와 인덱스 텐서를 가지고 있습니다. 인덱스 텐서의 인덱스에 해당하는 텐서의 값을 특정 스칼라로 설정하려고합니다. 어떻게해야합니까?Nensy의 배열에 해당하는 Tensorflow [인덱스] = 스칼라

array[indices] = scalar 

내 구체적인 경우에 우리가 1 차원 텐서에 대해 얘기하고 :

mask = tf.zeros_like(some_1D_tensor) 

(e.g. mask = [0, 0, 0, 0, 0]) 

하자 즉

, 나는 다음과 같은 NumPy와 작업의 Tensorflow에 해당 찾고 있어요 indices은 스칼라 값 1로 설정하고자하는 mask의 인덱스를 포함하는 1D 텐서가되어야합니다. 그래서 원하는 값 :

mask[indices] = 1 

(e.g. for indices = [1, 3] the output should be mask == [0, 1, 0, 1, 0]) 

답변

2

나는 그 전이나 난 그냥 그것을 볼 수없는 경우가 아니었다면 알고 있지만

array[indices] = scalar 

의 일반적인 경우에 해당하여

tensor = tf.scatter_nd_update(tensor, indices, updates) 

하지 않는 tf.scatter_nd_update()

관련 문제