Trả lời cho sự kiện TensorFlow Everywhere tại địa phương của bạn ngay hôm nay!
Trang này được dịch bởi Cloud Translation API.
Switch to English

dòng chảy :: ops :: Tụ họp

#include <array_ops.h>

Thu thập lát từ params theo indices .

Tóm lược

indices phải là một số nguyên có kích thước bất kỳ (thường là 0-D hoặc 1-D). Tạo ra một tensor đầu ra với shape 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] sau đó hoạt động này sẽ hoán vị params cho phù hợp.

validate_indices : DEPRECATED. Nếu thao tác này được gán cho CPU, các giá trị trong indices luôn được xác nhận là nằm trong phạm vi. Nếu được chỉ định cho GPU, các chỉ số không bị ràng buộc dẫn đến hành vi an toàn nhưng không xác định, có thể bao gồm việc phát sinh lỗi.

Tranh luận:

Lợi nhuận:

  • Output : Bộ căng đầu ra.

Người xây dựng và Người phá hủy

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)

Thuộc tính công cộng

operation
output

Chức năng công cộng

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

Chức năng tĩnh công khai

ValidateIndices (bool x)

Cấu trúc

tensorflow :: ops :: Gather :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho Tập hợp .

Thuộc tính công cộng

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

Chức năng công cộng

Tụ họp

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

Tụ họp

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

nút

::tensorflow::Node * node() const 

toán tử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const 

toán tử :: tensorflow :: Đầu ra

 operator::tensorflow::Output() const 

Chức năng tĩnh công khai

ValidateIndices

Attrs ValidateIndices(
  bool x
)