ML 커뮤니티 데이는 11월 9일입니다! TensorFlow, JAX에서 업데이트를 우리와 함께, 더 자세히 알아보기

tensorflow :: ops :: ScatterAdd

#include <state_ops.h>

변수 참조에 희소 업데이트를 추가합니다.

요약

이 연산은

# Scalar indices
ref[indices, ...] += updates[...]

# Vector indices (for each i)
ref[indices[i], ...] += updates[i, ...]

# High rank indices (for each i, ..., j)
ref[indices[i, ..., j], ...] += updates[i, ..., j, ...]
을 계산합니다.

이 작업은 업데이트가 완료된 후 ref 를 출력합니다. 이렇게하면 재설정 값을 사용해야하는 작업을 더 쉽게 연결할 수 있습니다.

중복 항목은 올바르게 처리됩니다. 여러 indices 가 동일한 위치를 참조하면 해당 항목이 추가됩니다.

updates.shape = indices.shape + ref.shape[1:] 또는 updates.shape = [] 합니다.

인수 :

  • 범위 : 범위 개체
  • ref : Variable 노드에서 Variable 합니다.
  • 인덱스 : ref 의 첫 번째 차원에 대한 인덱스 텐서.
  • updates : ref 에 추가 할 업데이트 된 값의 텐서.

선택적 속성 ( Attrs 참조) :

  • use_locking : True이면 추가가 잠금으로 보호됩니다. 그렇지 않으면 동작이 정의되지 않지만 경합이 적을 수 있습니다.

보고:

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

생성자와 소멸자

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

공용 속성

operation
output_ref

공공 기능

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

공개 정적 함수

UseLocking (bool x)

구조체

tensorflow :: ops :: ScatterAdd :: 속성

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

공용 속성

조작

Operation operation

output_ref

::tensorflow::Output output_ref

공공 기능

ScatterAdd

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

ScatterAdd

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

마디

::tensorflow::Node * node() const 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const 

공개 정적 함수

사용 잠금

Attrs UseLocking(
  bool x
)