Giảm bớt việc chuyển mã sử dụng tf.nn.embedding_lookup_sparse().
embedding_indices[i] và aggregation_weights[i] tương ứng với tính năng thứ i.
Các tensor tại các vị trí tương ứng trong ba danh sách đầu vào (sample_indices, embedding_indices và aggregation_weights) phải có hình dạng giống nhau, tức là xếp hạng 1 với dim_size() bằng tổng số lần tra cứu vào bảng được mô tả bởi đặc tính tương ứng.
Các lớp lồng nhau
| lớp học | EnqueueTPUEembeddingArbitraryTensorBatch.Options | Các thuộc tính tùy chọn cho EnqueueTPUEmbeddingArbitraryTensorBatch | |
Phương pháp công khai
| EnqueueTPUEmbeddingArbitraryTensorBatch.Options tĩnh | bộ kết hợp (Danh sách <String> bộ kết hợp) |
| static <T mở rộng Số, U mở rộng Số, V mở rộng Số> EnqueueTPUEmbeddingArbitraryTensorBatch | tạo ( Phạm vi phạm vi, Iterable< Toán hạng <T>> sampleIndicesOrRowSplits, Iterable< Toán hạng <U>> embeddingIndices, Iterable< Toán hạng <V>> tập hợp Trọng lượng, Chế độ toán hạng <String> Ghi đè, Tùy chọn... tùy chọn) Phương thức gốc để tạo một lớp bao bọc một thao tác EnqueueTPUEmbeddingArbitraryTensorBatch mới. |
| EnqueueTPUEmbeddingArbitraryTensorBatch.Options tĩnh | deviceOrdinal (Thiết bị dàiOrdinal) |
Phương pháp kế thừa
Phương pháp công khai
công khai tĩnh EnqueueTPUEmbeddingArbitraryTensorBatch.Options bộ kết hợp (Danh sách <String> bộ kết hợp)
Thông số
| máy kết hợp | Danh sách các chuỗi vô hướng, một chuỗi cho mỗi bảng nhúng chỉ định cách chuẩn hóa các kích hoạt nhúng sau khi tính tổng có trọng số. Các bộ kết hợp được hỗ trợ là 'trung bình', 'tổng' hoặc 'sqrtn'. Sẽ không hợp lệ nếu tổng trọng số bằng 0 đối với 'trung bình' hoặc tổng trọng số bình phương là 0 đối với 'sqrtn'. Nếu bộ kết hợp không được thông qua thì mặc định là sử dụng 'tổng' cho tất cả các bảng. |
|---|
public static EnqueueTPUEmbeddingArbitraryTensorBatch create ( Phạm vi phạm vi, Iterable< Operand <T>> sampleIndicesOrRowSplits, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... options)
Phương thức gốc để tạo một lớp bao bọc một thao tác EnqueueTPUEmbeddingArbitraryTensorBatch mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| mẫuChỉ sốOrRowSplits | Danh sách Tensors cấp 2 chỉ định ví dụ huấn luyện chứa các giá trị embedding_indices và aggregation_weights tương ứng. Nếu kích thước của thứ nguyên đầu tiên là 0 thì chúng tôi giả sử mỗi embedding_indices thuộc về một mẫu khác nhau. Cả int32 và int64 đều được phép và sẽ được chuyển đổi nội bộ thành int32. Hoặc danh sách Tensors xếp hạng 1 chỉ định cách phân chia hàng để chia embedding_indices và aggregation_weights thành hàng. Nó tương ứng với ids.row_splits trong embedding_lookup(), khi id là RaggedTensor. Khi xếp hàng tenxơ rời rạc ND vào hàng đợi, chỉ có chiều cuối cùng được phép rời rạc. sự phân chia hàng là tensor dày đặc 1-D. Khi trống, chúng tôi giả sử một tensor dày đặc được truyền tới op. Cả int32 và int64 đều được phép và sẽ được chuyển đổi thành int32 bên trong. |
| nhúngChỉ số | Danh sách Tensors hạng 1, chỉ số vào bảng nhúng. Cả int32 và int64 đều được phép và sẽ được chuyển đổi nội bộ thành int32. |
| tổng hợpTrọng lượng | Danh sách các Tensor xếp hạng 1 chứa các trọng số tổng hợp cho mỗi ví dụ huấn luyện. Cả float32 và float64 đều được phép và sẽ được chuyển đổi thành float32 trong nội bộ. |
| chế độghi đè | Đầu vào chuỗi ghi đè chế độ được chỉ định trong TPUEmbeddingConfiguration. Các giá trị được hỗ trợ là {'unspecified', 'inference', 'training', 'backward_pass_only'}. Khi được đặt thành 'không xác định', chế độ được đặt trong TPUEmbeddingConfiguration sẽ được sử dụng, nếu không thì mode_override sẽ được sử dụng. |
| 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 EnqueueTPUEmbeddingArbitraryTensorBatch
public static EnqueueTPUEmbeddingArbitraryTensorBatch.Options deviceOrdinal (Long deviceOrdinal)
Thông số
| thiết bịthứ tự | Thiết bị TPU sẽ sử dụng. Phải >= 0 và nhỏ hơn số lõi TPU trong tác vụ mà nút được đặt trên đó. |
|---|