lớp cuối cùng công khai XlaRecvTPUEmbeddingActivations
Một op nhận các kích hoạt nhúng trên TPU.
Hệ thống TPU thực hiện việc tra cứu và tổng hợp nhúng. Kết quả của những sự tổng hợp này được hiển thị trên Biểu đồ Tensorflow dưới dạng đầu ra của XlaRecvTPUEmbeddingActivations Op. Op này trả về một danh sách chứa một Tensor kích hoạt trên mỗi bảng được chỉ định trong mô hình.
Phương pháp công khai
| XlaRecvTPUEnhúngKích hoạt tĩnh | tạo ( Phạm vi phạm vi, Toán hạng <?> Chống trùng lặpData, numTables dài, Cấu hình chuỗi) Phương thức ban đầu để tạo một lớp bao bọc hoạt động XlaRecvTPUEmbeddingActivations mới. |
| Trình vòng lặp< Toán hạng <Float>> | trình lặp () |
| Danh sách< Đầu ra <Float>> | đầu ra () Một TensorList bao gồm các kích hoạt nhúng chứa một Tensor trên mỗi bảng nhúng trong mô hình. |
Phương pháp kế thừa
Phương pháp công khai
công khai tĩnh XlaRecvTPUEmbeddingActivations tạo ( Phạm vi phạm vi, Toán hạng <?> chống trùng lặpData, numTables dài, Cấu hình chuỗi)
Phương thức ban đầu để tạo một lớp bao bọc hoạt động XlaRecvTPUEmbeddingActivations mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| chống trùng lặp dữ liệu | Một Tensor có type=DT_VARIANT chứa dữ liệu chống trùng lặp. Tensor là một bộ dữ liệu lồng nhau XLA chứa N phần tử (trong đó N là tỷ lệ của số lần nhúng vào lõi tensor trên mỗi chip TPU). Mỗi phần tử của bộ dữ liệu lồng nhau là một bộ tenxơ hạng 1. Mỗi tensor chứa các chỉ số (DT_UINT32) để nhúng tra cứu trên TensorCore hoặc chứa các trọng số (DT_FLOAT) để áp dụng cho đầu ra của thao tác tra cứu nhúng. |
| số bảng | Số lượng tensor kích hoạt đầu ra. Nếu bộ mô tả tính năng có trong cấu hình nhúng tpu, thì nó bằng số lượng tính năng, nếu không thì bằng số lượng bảng nhúng trong mô hình. |
| cấu hình | Proto Cấu hình TPUEmbedding được tuần tự hóa. |
Trả lại
- một phiên bản mới của XlaRecvTPUEmbeddingActivations
Danh sách công khai< Đầu ra <Float>> đầu ra ()
Một TensorList bao gồm các kích hoạt nhúng chứa một Tensor trên mỗi bảng nhúng trong mô hình.