lớp cuối cùng công khai EnqueueTPUEmbeddingBatch
Một op sắp xếp danh sách các tensor lô đầu vào vào TPUEmbedding.
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 | EnqueueTPUEmbeddingBatch.Options | Thuộc tính tùy chọn cho EnqueueTPUEmbeddingBatch | |
Phương pháp công khai
| EnqueueTPUEmbeddingBatch.Options tĩnh | bộ kết hợp (Danh sách <String> bộ kết hợp) |
| EnqueueTPUEembeddingBatch tĩnh | tạo ( Phạm vi phạm vi, Iterable< Operand <String>> batch, Operand <String> modeOverride, 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 EnqueueTPUEmbeddingBatch mới. |
| EnqueueTPUEmbeddingBatch.Options tĩnh | deviceOrdinal (Thiết bị dàiOrdinal) |
Phương pháp kế thừa
Phương pháp công khai
bộ kết hợp EnqueueTPUEmbeddingBatch.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 EnqueueTPUEmbeddingBatch create (Phạm vi phạm vi , Iterable< Toán hạng <String>> batch, 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 EnqueueTPUEmbeddingBatch mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| lô hàng | Một danh sách các tensor 1D, một cho mỗi bảng nhúng, chứa các đầu vào hàng loạt được mã hóa dưới dạng các giao thức dist_belief.SparseFeatures. Nếu trường trọng số trong giao thức SparseFeatures không được điền cho ID thì giả sử trọng số là 1.0. |
| 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 EnqueueTPUEmbeddingBatch
public static EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
Thông số
| thiết bịthứ tự | Thiết bị TPU sẽ sử dụng. Giá trị này phải là -1 khi Op đang chạy trên thiết bị TPU và >= 0 khi Op đang chạy trên thiết bị CPU. |
|---|