lớp cuối cùng công khai EnqueueTPUEmbeddingSparseBatch
Một hoạt động xếp hàng các chỉ số đầu vào TPUEmbedding từ SparseTensor.
Op này giúp dễ dàng chuyển mã sử dụng embedding_lookup_sparse(), mặc dù cần phải xử lý trước một số Python đối số SparseTensor thành embedding_lookup_sparse() để tạo đối số cho Op này, vì chỉ cho phép một EnqueueTPUEmbeddingSparseBatch Op duy nhất cho mỗi bước đào tạo.
Các tensor tại các vị trí tương ứng trong 3 danh sách đầu vào phải có hình dạng giống nhau, tức là 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 table_id tương ứng.
Các lớp lồng nhau
| lớp học | EnqueueTPUEembeddingSparseBatch.Options | Các thuộc tính tùy chọn cho EnqueueTPUEmbeddingSparseBatch | |
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 khai
| EnqueueTPUEmbeddingSparseBatch.Options tĩnh | bộ kết hợp (Danh sách <String> bộ kết hợp) |
| static <T mở rộng TNumber , U mở rộng TNumber , V mở rộng TNumber > EnqueueTPUEmbeddingSparseBatch | |
| EnqueueTPUEmbeddingSparseBatch.Options tĩnh | deviceOrdinal (Thiết bị dàiOrdinal) |
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: "EnqueueTPUEmbeddingSparseBatch"
Phương pháp công khai
bộ kết hợp EnqueueTPUEmbeddingSparseBatch.Options tĩnh công khai (Bộ kết hợp Danh sách<String>)
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 EnqueueTPUEmbeddingSparseBatch create ( Phạm vi phạm vi, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand < TString > modeOverride, Options... options)
Phương thức gốc để tạo một lớp bao bọc một thao tác EnqueueTPUEmbeddingSparseBatch mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| mẫuChỉ số | Danh sách Tensors xếp hạng 1 chỉ định ví dụ huấn luyện và tính năng chứa các giá trị embedding_indices và aggregation_weights tương ứng. sample_indices[i] phải bằng b * nf + f, trong đó nf là số đối tượng từ bảng tương ứng, f nằm trong [0, nf) và b nằm trong [0, cỡ lô). |
| nhúngChỉ số | Danh sách Tensors hạng 1, chỉ số vào bảng nhúng. |
| tổng hợpTrọng lượng | Danh sách Tensor xếp hạng 1 chứa mỗi mẫu -- tức là trên mỗi (ví dụ huấn luyện, tính năng) -- trọng số tổng hợp. |
| 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 EnqueueTPUEmbeddingSparseBatch
public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Thiết bị dàiOrdinal)
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 đó. |
|---|