텐서플로우:: ops:: ScatterNdSub
#include <state_ops.h>변수 의 개별 값 또는 슬라이스에 희소 뺄셈을 적용합니다.
요약
주어진 변수 내에서 indices 에 따라.
ref 는 랭크가 P 인 Tensor 이고 indices 랭크가 Q 인 Tensor 입니다.
indices ref 에 대한 인덱스를 포함하는 정수 텐서여야 합니다. 모양은 [d_0, ..., d_{Q-2}, K] 이어야 하며, 여기서 0 < K <= P .
indices 의 가장 안쪽 차원(길이 K )은 ref 의 K 번째 차원을 따라 요소( K = P 인 경우) 또는 슬라이스( K < P 인 경우)에 대한 인덱스에 해당합니다.
updates 는 Q-1+PK 랭크의 Tensor 이며 형태는 다음과 같습니다.
[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]
예를 들어, 8개의 요소를 가진 랭크 1 텐서에서 흩어져 있는 4개의 요소를 빼고 싶다고 가정해 보겠습니다. 파이썬에서 이 뺄셈은 다음과 같이 표현됩니다.
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]) sub = tf.scatter_nd_sub(ref, indices, updates) with tf.Session() as sess: print sess.run(sub)
ref에 대한 업데이트 결과는 다음과 같습니다.
[1, -9, 3, -6, -4, 6, 7, -4]
슬라이스를 업데이트하는 방법에 대한 자세한 내용은 tf.scatter_nd 참조하세요.
논거:
- 범위: Scope 객체
- 참고: 변경 가능한 텐서입니다 . Variable 노드에서 가져와야 합니다.
- indices: 텐서입니다 . int32 또는 int64 유형 중 하나여야 합니다. ref에 대한 인덱스 텐서입니다.
- 업데이트: 텐서입니다 . 참조(ref)와 동일한 유형이어야 합니다. 참조에서 뺄 업데이트된 값의 텐서입니다.
선택적 속성( Attrs 참조):
- use_locking: 선택적 부울 값입니다. 기본값은 True입니다. True로 설정하면 할당 작업이 락으로 보호됩니다. 그렇지 않으면 동작은 정의되지 않지만, 경합이 줄어들 수 있습니다.
보고:
-
Output: 참조 값과 동일합니다. 업데이트가 완료된 후 업데이트된 값을 사용하려는 작업의 편의를 위해 반환됩니다.
생성자와 소멸자 | |
|---|---|
ScatterNdSub (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ScatterNdSub (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterNdSub::Attrs & attrs) |
공개 속성 | |
|---|---|
operation | |
output_ref | |
공공 기능 | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
공개 정적 함수 | |
|---|---|
UseLocking (bool x) | |
구조 | |
|---|---|
| 텐서플로우:: ops:: ScatterNdSub:: Attrs | ScatterNdSub 에 대한 선택적 속성 설정기입니다. |
공개 속성
작업
Operation operation
출력 참조
::tensorflow::Output output_ref
공공 기능
스캐터NdSub
ScatterNdSub( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
스캐터NdSub
ScatterNdSub( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterNdSub::Attrs & attrs )
마디
::tensorflow::Node * node() const
연산자::텐서플로우::입력
operator::tensorflow::Input() const
연산자::텐서플로우::출력
operator::tensorflow::Output() const
공개 정적 함수
잠금 사용
Attrs UseLocking( bool x )