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]
![](https://tensorflow.google.cn/images/Gather.png?authuser=5&hl=vi)
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 |
Phương pháp công khai
Đầu ra <T> | asOutput () Trả về phần điều khiển tượng trưng của một tenxơ. |
Tập hợp tĩnh.Options | batchDims (Đợt dài) |
tĩnh <T, U mở rộng Số, V mở rộng Số> 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
Phương pháp công khai
Đầu ra công khai <T> asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
Đầ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>, chỉ số Toán hạng <U>, Trục Toán tử <V>, 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:]`.