Một op lấy các tham số tối ưu hóa từ việc nhúng vào bộ nhớ máy chủ.
Một op lấy các tham số tối ưu hóa từ việc nhúng vào bộ nhớ máy chủ. Trước đó phải có op configureTPUEmbeddingHost để thiết lập cấu hình bảng nhúng chính xác. Ví dụ: op này được sử dụng để truy xuất các tham số đã cập nhật trước khi lưu điểm kiểm tra. Đối với Adagrad, phụ trợ1 sẽ chứa các bộ tích lũy sau khi chạy op này. Đối với SGD, tất cả các giá trị phụ* sẽ trống (các tensor 0x0 cho bảng đó). Đối với FTRL, phần phụ1 sẽ chứa các bộ tích lũy và phần phụ2 sẽ chứa các thuật ngữ tuyến tính. Đối với ADAM, phụ trợ1 sẽ chứa mômen và phụ trợ2 sẽ chứa vận tốc.
Phương pháp công cộng
| Danh sách< Đầu ra <Float>> | phụ1 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ đầu tiên được lưu trữ. |
| Danh sách< Đầu ra <Float>> | phụ2 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ hai được lưu trữ. |
| Danh sách< Đầu ra <Float>> | phụ3 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ ba được lưu trữ. |
| Danh sách< Đầu ra <Float>> | phụ trợ4 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ tư được lưu trữ. |
| Danh sách< Đầu ra <Float>> | phụ5 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ năm được lưu trữ. |
| Danh sách< Đầu ra <Float>> | phụ6 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa sáu tham số tối ưu hóa phụ trợ được lưu trữ. |
| Danh sách< Đầu ra <Float>> | phụ7 () Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ bảy được lưu trữ. |
| tĩnh RetrieveAllTPUEembeddingParameters | |
| Danh sách< Đầu ra <Float>> | thông số () Một danh sách các tensor, một tensor cho mỗi bảng nhúng, chứa các tham số của bảng nhúng được lưu trữ. |
Phương pháp kế thừa
Phương pháp công cộng
Danh sách công khai< Đầu ra <Float>> phụ trợ1 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ đầu tiên được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
Danh sách công khai< Đầu ra <Float>> phụ trợ2 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ hai được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
Danh sách công khai< Đầu ra <Float>> phụ trợ3 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ ba được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
Danh sách công khai< Đầu ra <Float>> phụ trợ4 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ tư được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
Danh sách công khai< Đầu ra <Float>> phụ trợ5 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ năm được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
Danh sách công khai< Đầu ra <Float>> phụ trợ6 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa sáu tham số tối ưu hóa phụ trợ được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
Danh sách công khai< Đầu ra <Float>> phụ trợ7 ()
Một danh sách các tensor, một cho mỗi bảng nhúng, chứa tham số tối ưu hóa phụ trợ thứ bảy được lưu trữ. Các phần tử có trong danh sách nhưng có kích thước bằng 0, dành cho các tham số tối ưu hóa không được sử dụng (dựa trên thuật toán được sử dụng cho mỗi bảng).
tĩnh công khai RetrieveAllTPUEmbeddingParameters tạo (Phạm vi phạm vi , NumTables dài, Cấu hình chuỗi, numShards dài, shardId dài)
Phương thức ban đầu để tạo một lớp bao gồm một thao tác RetrieveAllTPUEmbeddingParameters mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| Số bảng | Số lượng bảng nhúng. |
| cấu hình | Một nguyên mẫu TPUEmbeddingConfiguration mô tả các tham số bảng đang được tải, được tuần tự hóa thành một chuỗi. |
| số mảnh vỡ | Số lượng phân đoạn mà các bảng nhúng được chia vào. |
| Id mảnh | Mã định danh phân đoạn cho hoạt động này. |
Trả lại
- một phiên bản mới của RetrieveAllTPUEmbeddingParameters
Danh sách công khai< Thông số đầu ra <Float>> ()
Một danh sách các tensor, một tensor cho mỗi bảng nhúng, chứa các tham số của bảng nhúng được lưu trữ.