Gather

lớp học cuối cùng công khai Tập hợp

Bao bọc toán tử XLA Gather được ghi lại tại

https://www.tensorflow.org/xla/Operation_semantics#gather

Hằng số

Sợi dây OP_NAME Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Phương pháp công cộng

Đầu ra <T>
asOutput ()
Trả về tay cầm tượng trưng của tensor.
tĩnh <T mở rộng TType , U mở rộng TNumber > Tập hợp <T>
tạo (Phạm vi phạm vi , toán hạng <T>, toán hạng <U> startIndices, toán hạng <U> sliceSizes, số thứ nguyên chuỗi, chỉ số Boolean được sắp xếp)
Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động Thu thập mới.
Đầu ra <T>
đầu ra ()

Phương pháp kế thừa

Hằng số

Chuỗi cuối cùng tĩnh công khai OP_NAME

Tên của op này, được biết đến bởi công cụ lõi TensorFlow

Giá trị không đổi: "XlaGather"

Phương pháp công cộng

Đầu ra công khai <T> asOutput ()

Trả về tay cầm tượng trưng của tensor.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

tĩnh công khai Thu thập <T> tạo ( Phạm vi phạm vi , toán hạng <T>, toán hạng <U> startIndices, toán hạng <U> sliceSizes, số thứ nguyên chuỗi, chỉ số Boolean được sắp xếp)

Phương thức xuất xưởng để tạo một lớp bao bọc một hoạt động Thu thập mới.

Thông số
phạm vi phạm vi hiện tại
toán hạng Mảng chúng tôi đang thu thập từ đó.
bắt đầuChỉ số Mảng chứa các chỉ số bắt đầu của các lát cắt mà chúng tôi thu thập được.
látKích thước slice_sizes[i] là giới hạn của lát cắt trên chiều thứ i.
số thứ nguyên Một nguyên mẫu xla::GatherDimensionNumbers được tuần tự hóa.
chỉ sốĐược sắp xếp Boolean cho biết các chỉ số đã được sắp xếp hay chưa.
Trả lại
  • một phiên bản mới của Gather

Đầu ra công khai đầu ra <T> ()