Tập hợp các lát cắt từ trục `params` theo `chỉ số`.
`chỉ số` 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 dạng `params.shape[:axis] +index.shape[batch_dims:] + params.shape[axis + 1:]` trong đó:
# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
Lưu ý rằng trên CPU, nếu tìm thấy chỉ mục ngoài giới hạn, lỗi sẽ được trả về. Trên GPU, nếu tìm thấy chỉ mục ngoài giới hạn, số 0 sẽ được lưu trong giá trị đầu ra tương ứng.
Xem thêm `tf.batch_gather` và tf.gather_nd .
Các lớp lồng nhau
| lớp học | Tập hợp.Options | Thuộc tính tùy chọn cho 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 khai
| Đầu ra <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
| Tập hợp tĩnh.Options | batchDims (Đợt dài) |
| tĩnh <T mở rộng TType > Tập hợp <T> | |
| Đầu ra <T> | đầu ra () Các giá trị từ `params` được thu thập từ các chỉ mục do `indices` cung cấp, có hình dạng `params.shape[:axis] +index.shape + params.shape[axis + 1:]`. |
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
Phương pháp công khai
Đầ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.
public static Tập hợp <T> tạo ( Phạm vi phạm vi, thông số Toán hạng <T>, Toán hạng <? mở rộng TNumber > chỉ mục, Toán hạng <? mở rộng TNumber > trục, Tùy chọn... tùy chọn)
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 |
|---|---|
| thông số | Tenxor để thu thập giá trị. Ít nhất phải có thứ hạng `trục + 1`. |
| chỉ số | Tenxơ chỉ số. Phải nằm trong phạm vi `[0, params.shape[axis])`. |
| trục | Trục trong `params` để thu thập `chỉ số` từ đó. Mặc định là kích thước đầu tiên. Hỗ trợ các chỉ số tiêu cực. |
| tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của Gather
Đầu ra công khai đầu ra <T> ()
Các giá trị từ `params` được thu thập từ các chỉ mục do `indices` cung cấp, có hình dạng `params.shape[:axis] +index.shape + params.shape[axis + 1:]`.