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]
trong đó- `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.
Phương pháp công khai
static <T mở rộng Số, U, V mở rộng Số> 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
Phương pháp công khai
public static RaggedGather <T, U> create ( Phạm vi phạm vi, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> chỉ số, Long OUTPUTRAGGEDRANK)
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ề.