XlaRecvTPUEmbeddingDeduplicationData

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

Nhận dữ liệu chống trùng lặp (chỉ số và trọng số) từ lõi nhúng.

Dữ liệu chống trùng lặp là một Tensor có loại=DT_VARIANT. Bản thân 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.

Phương pháp công cộng

Đầu ra <Đối tượng>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
tĩnh XlaRecvTPUEnhúngDduplicationData
tạo ( Phạm vi phạm vi, Cấu hình chuỗi)
Phương thức ban đầu để tạo một lớp bao gồm thao tác XlaRecvTPUEmbeddingDeduplicationData mới.
Đầu ra <?>
đầu ra ()

Phương pháp kế thừa

Phương pháp công cộng

Đầu ra công khai <Object> asOutput ()

Trả về phần điều khiển tượng trưng của một tenxơ.

Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.

public static XlaRecvTPUEmbeddingDeduplicationData create ( Phạm vi phạm vi, Cấu hình chuỗi)

Phương thức ban đầu để tạo một lớp bao gồm thao tác XlaRecvTPUEmbeddingDeduplicationData mới.

Thông số
phạm vi phạm vi hiện tại
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 XlaRecvTPUEmbeddingDeduplicationData

Đầu ra công khai <?> đầu ra ()