dòng chảy căng:: ôi:: Tập trung

Tập hợp các lát từ params theo indices .

Bản tóm tắt

indices phải là một tensor nguyên có chiều bất kỳ (thường là 0-D hoặc 1-D). Tạo ra một tenxơ đầu ra có hình indices.shape + params.shape[1:] trong đó:

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

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

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

Nếu indices là một hoán vị và len(indices) == params.shape[0] thì thao tác này sẽ hoán vị params tương ứng.

validate_indices : KHÔNG DÙNG. Nếu thao tác này được gán cho CPU, các giá trị trong indices luôn được xác thực nằm trong phạm vi. Nếu được gán cho GPU, các chỉ số nằm ngoài giới hạn sẽ dẫn đến hành vi an toàn nhưng không xác định, có thể bao gồm cả việc gây ra lỗi.

Lập luận:

• phạm vi: Một đối tượng phạm vi

Trả về:

• Output : Tenxor đầu ra.

Thuộc tính công khai

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
)