Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add a layer to dynamically generate mask #54

Closed
chenmoneygithub opened this issue Mar 22, 2022 · 1 comment
Closed

Add a layer to dynamically generate mask #54

chenmoneygithub opened this issue Mar 22, 2022 · 1 comment
Assignees

Comments

@chenmoneygithub
Copy link
Contributor

Is your feature request related to a problem? Please describe.
Masking is an essential part in LM models, so we need to have a layer to do the work.

Describe the solution you'd like
Create a new layer, which takes in sequence data (could be both RaggedTensor and Tensor), and returns masked sequence, and masked positions and ids.

Describe alternatives you've considered
We can directly use tf_text, but the API is not very user-friendly.

Additional context
N/A

@mattdangerw
Copy link
Member

This is done! #61

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants