lớp cuối cùng công khai EnqueueTPUEmbeddingIntegerBatch
Một op sắp xếp danh sách các tensor lô đầu vào vào TPUEmbedding.
Các lớp lồng nhau
| lớp học | EnqueueTPUEmbeddingIntegerBatch.Options | Thuộc tính tùy chọn cho EnqueueTPUEmbeddingIntegerBatch | |
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
| EnqueueTPUEembeddingIntegerBatch tĩnh | |
| EnqueueTPUEmbeddingIntegerBatch.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: "EnqueueTPUEmbeddingIntegerBatch"
Phương pháp công khai
public static EnqueueTPUEmbeddingIntegerBatch create ( Phạm vi phạm vi, Iterable< Operand < TInt32 >> batch, 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 EnqueueTPUEmbeddingIntegerBatch mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| lô hàng | Danh sách các tensor 1D, một cho mỗi bảng nhúng, chứa các chỉ số trong các bảng. |
| 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 EnqueueTPUEmbeddingIntegerBatch
public static EnqueueTPUEmbeddingIntegerBatch.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 đó. |
|---|