lớp cuối cùng công khai RaggedGather
Tập hợp các lát cắt rời rạc từ trục `params` `0` theo `chỉ số`.
Xuất ra một đầu ra `RaggedTensor` được tạo từ `output_dense_values` và `output_nested_splits`, sao cho:
output.shape = indices.shape + params.shape[1:]
output.ragged_rank = indices.shape.ndims + params.ragged_rank
output[i...j, d0...dn] = params[indices[i...j], d0...dn]
- `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` cung cấp các giá trị cần được thu thập.
- `chỉ số` là một tensor dày đặc với dtype `int32` hoặc `int64`, cho biết giá trị nào sẽ được thu thập.
- `output = ragged.from_nested_row_splits(output_dense_values, out_nested_splits)` là tensor đầu ra.
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
| tĩnh <T mở rộng TNumber , U mở rộng TType > RaggedGather <T, U> | |
| Đầu ra <U> | đầu raDenseValues () `flat_values` cho RaggedTensor được trả về. |
| Danh sách< Đầu ra <T>> | đầu raNestedSplits () Các tensor `nested_row_splits` xác định việc phân vùng hàng cho RaggedTensor được trả về. |
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: "RaggedGather"
Phương pháp công cộng
public static RaggedGather <T, U> tạo ( Phạm vi phạm vi, Iterable< Toán hạng <T>> paramsNestedSplits, Toán hạng <U> paramsDenseValues, Toán hạng <? mở rộng TNumber > chỉ số, OUTPUTRAGGEDRANK dài)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác RaggedGather mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| thông sốNestedSplits | Các tensor `nested_row_splits` xác định việc phân vùng hàng cho đầu vào RaggedTensor của `params`. |
| thông sốDenseValues | `giá trị phẳng` cho `params` RaggedTensor. Đã có sự thay đổi về thuật ngữ ở cấp độ python từdense_values thành Flat_values, vì vậy,dense_values là tên không được dùng nữa. |
| chỉ số | Các chỉ số ở chiều ngoài cùng của `thông số` của các giá trị cần được thu thập. |
| OUTPUTRAGGEDRANK | Thứ hạng rách rưới của RaggedTensor đầu ra. `output_nested_splits` sẽ chứa số tensor `row_splits` này. Giá trị này phải bằng `indices.shape.ndims + params.ragged_rank - 1`. |
Trả lại
- một phiên bản mới của RaggedGather
Danh sách công khai< Đầu ra <T>> đầu raNestedSplits ()
Các tensor `nested_row_splits` xác định việc phân vùng hàng cho RaggedTensor được trả về.