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])