텐서플로우:: 작전:: 모으다

indices 에 따라 params 에서 조각을 수집합니다 .

요약

indices 모든 차원(보통 0-D 또는 1-D)의 정수 텐서여야 합니다. indices.shape + params.shape[1:] 형태의 출력 텐서를 생성합니다. 여기서:

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

indices 순열이고 len(indices) == params.shape[0] 인 경우 이 작업은 그에 따라 params 순열합니다.

validate_indices : 더 이상 사용되지 않습니다. 이 작업이 CPU에 할당되면 indices 의 값은 항상 범위 내에 있는지 검증됩니다. GPU에 할당된 경우 범위를 벗어난 인덱스는 안전하지만 지정되지 않은 동작을 발생시키며, 여기에는 오류 발생이 포함될 수 있습니다.

인수:

• 범위: 범위 개체

보고:

• Output : 출력 텐서.

공개 속성

Operation operation

::tensorflow::Output output

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices
)

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  const Gather::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs ValidateIndices(
  bool x
)