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 được chỉ định bởi các đối số cho TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch. 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 hoạt động RecvTPUEmbeddingActivations. 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. Có thể có nhiều nhất một hoạt động RecvTPUEmbeddingActivations trong biểu đồ TPU.
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
Kích hoạt RecvTPUEnhúng tĩnh | |
Trình vòng lặp< Toán hạng < TFloat32 >> | trình lặp () |
Danh sách< Đầu ra < TFloat32 >> | đầ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
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
Phương pháp công khai
Tạo RecvTPUEmbeddingActivations tĩnh công khai (Phạm vi phạm vi , Số đầu ra dài, Cấu hình chuỗi)
Phương thức gốc để tạo một lớp bao bọc một thao tác RecvTPUEmbeddingActivations mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
số kết quả đầu ra | Số lượng tensor kích hoạt đầu ra, 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 RecvTPUEmbeddingActivations