RaggedGather

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]
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.
(Lưu ý: op c++ ​​này được sử dụng để triển khai op python `tf.ragged.gather` cấp cao hơn, cũng hỗ trợ các chỉ mục rời rạc.)

Phương pháp công khai

static <T mở rộng Số, U, V mở rộng Số> RaggedGather <T, U>
tạo ( Phạm vi phạm vi, Iterable< Toán hạng <T>> paramsNestedSplits, Toán hạng <U> paramsDenseValues, Chỉ số toán hạng <V>, 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.
Đầ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

Đầu ra công khai <U> đầu raDenseValues ​​()

`flat_values` cho RaggedTensor được trả về.

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ề.