RecvTPUEmbeddingActivations

lớp cuối cùng công khai RecvTPUEembeddingActivations

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.

Phương pháp công khai

Kích hoạt RecvTPUEnhúng tĩnh
tạo ( 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.
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

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

public Iterator< Toán hạng <Float>> iterator ()

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.