TensorFlow에서 Tensor 항목을 지우는 방법은 무엇입니까?

Question

몇 가지 계산 후에 Tensor x 모양 (128, 128, 32)을 얻습니다. 다음 코드에서와 동일한 동작을 어떻게 얻을 수 있습니까?

x[:, :, 1:] = 0

실제로, 나는 하나의 매트릭스를 제외한 모든 항목을 삭제하고 싶습니다.

텐서는 세션 내에서 얻어지며 동일한 세션에서 항목을 지우고 싶습니다.

가장 쉬운 방법은 무엇입니까?

Answer 1

0.10부터 TensorFlow에는 이러한 용도로 사용할 수있는 고급 인덱싱이 없으므로 벡터 모양을 변경하고 선형 인덱스 집합에 할당 한 다음 다시 모양을 다시 잡아야합니다. 변수 + scatter_update 또는 dynamic_stitch를 지정하여 벡터에서이 할당을 수행 할 수 있습니다. here