lớp tĩnh công khai EnqueueTPUEmbeddingSparseBatch.Options
Các thuộc tính tùy chọn cho EnqueueTPUEmbeddingSparseBatch
Phương pháp công khai
EnqueueTPUEembeddingSparseBatch.Options | bộ kết hợp (Danh sách <String> bộ kết hợp) |
EnqueueTPUEembeddingSparseBatch.Options | 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 EnqueueTPUEmbeddingSparseBatch.Options 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 EnqueueTPUEembeddingSparseBatch.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 đó. |
---|