Google I/O에 참여해 주셔서 감사합니다. 주문형 모든 세션 보기 주문형 시청

tensorflow :: ops :: ScatterNdUpdate

#include <state_ops.h>

주어진 내의 개별 값이나 조각에 희소 updates 를 적용 updates .

요약

indices 에 따라 변수.

ref 는 순위가 P Tensor 이고 indices 는 순위가 Q Tensor 입니다.

indicesref 인덱스를 포함하는 정수 텐서 여야합니다. 0 < K <= P \([d_0, ..., d_{Q-2}, K]\) 모양이어야합니다.

indices 의 가장 안쪽 차원 (길이 K )은 refK 번째 차원을 따라 요소 ( K = P ) 또는 슬라이스 ( K < P 경우)에 대한 인덱스에 해당합니다.

updatesQ-1+PK 등급의 Tensor 입니다.

$$[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]].$$

예를 들어 4 개의 흩어진 요소를 랭크 1 텐서로 업데이트하고 8 개 요소로 업데이트한다고 가정 해 보겠습니다. Python에서 해당 업데이트는 다음과 같습니다.

    ref = tf.Variable([1, 2, 3, 4, 5, 6, 7, 8])
    indices = tf.constant([[4], [3], [1] ,[7]])
    updates = tf.constant([9, 10, 11, 12])
    update = tf.scatter_nd_update(ref, indices, updates)
    with tf.Session() as sess:
      print sess.run(update)

ref에 대한 결과 업데이트는 다음과 같습니다.

[1, 11, 3, 10, 9, 6, 7, 12]

슬라이스를 업데이트하는 방법에 대한 자세한 내용은 tf.scatter_nd 를 참조하세요.

tf.scatter_updatetf.batch_scatter_update 도 참조하십시오.

인수 :

  • 범위 : 범위 개체
  • ref : 가변 Tensor . 변수 노드에서 가져와야 합니다.
  • 인덱스 : Tensor . 다음 유형 중 하나 여야합니다 : int32, int64. ref에 대한 인덱스의 텐서.
  • 업데이트 : Tensor . ref와 동일한 유형이어야합니다. ref에 추가 할 업데이트 된 값의 텐서입니다.

선택적 속성 ( Attrs 참조) :

  • use_locking : 선택적 bool입니다. 기본값은 True입니다. True이면 할당이 잠금으로 보호됩니다. 그렇지 않으면 동작이 정의되지 않지만 경합이 적을 수 있습니다.

보고:

  • Output : ref. 업데이트가 완료된 후 업데이트 된 값을 사용하려는 작업의 편의를 위해 반환됩니다.

생성자와 소멸자

ScatterNdUpdate (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates)
ScatterNdUpdate (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterNdUpdate::Attrs & attrs)

공용 속성

operation
output_ref

공공 기능

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

공개 정적 함수

UseLocking (bool x)

구조체

tensorflow :: ops :: ScatterNdUpdate :: Attrs

ScatterNdUpdate에 대한 선택적 속성 설정자.

공용 속성

조작

Operation operation

output_ref

::tensorflow::Output output_ref

공공 기능

ScatterNdUpdate

 ScatterNdUpdate(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates
)

ScatterNdUpdate

 ScatterNdUpdate(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates,
  const ScatterNdUpdate::Attrs & attrs
)

마디

::tensorflow::Node * node() const 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const 

공개 정적 함수

사용 잠금

Attrs UseLocking(
  bool x
)