Định nghĩa mô hình team

Bộ tra cứu sức mạnh team có ba thành phần: một đầu vào, một lớp embedding, và một lớp flatten để tạo đầu ra.

Nếu bạn gói ba lớp này trong một mô hình với một đầu vào và một đầu ra, bạn có thể tái sử dụng cụm ba lớp đó ở nhiều nơi.

Lưu ý thêm rằng trọng số của cả ba lớp sẽ được chia sẻ ở mọi nơi chúng ta dùng chúng.

Tạo một lớp đầu vào 1D cho team ID (sẽ là một số nguyên). Nhớ đặt đúng shape đầu vào!
Truyền đầu vào này vào lớp tra cứu sức mạnh team mà bạn đã tạo trước đó.
Làm phẳng (flatten) đầu ra của lớp tra cứu sức mạnh team.
Tạo một mô hình dùng đầu vào 1D làm input và sức mạnh team đã được làm phẳng làm output.