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]
Ở đâu
  • `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.)

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>
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 <? kéo dài 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.
Đầ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_valueslà 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ề.