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

support jagged tensor with no inner dense dim #1267

Closed
wants to merge 1 commit into from

Conversation

jspark1105
Copy link
Contributor

Summary:
jagged_to_padded_dense supports jagged tensor with inner dense dim is 1 and folded like the following example, but producing the error x_offsets.size(), 1 != num_jagged_dim, 0 in backward.

x_values = [5, 3, 7, 1]
x_offsets = [0, 1, 4]
max_length = 3
->
x_padded = [[5, 0, 0], [3, 7, 1]]

Differential Revision: D39121072

@netlify
Copy link

netlify bot commented Aug 30, 2022

Deploy Preview for pytorch-fbgemm-docs canceled.

Name Link
🔨 Latest commit a535ff5
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/63b1d6be44c5d10008d5c495

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D39121072

Summary:
Pull Request resolved: pytorch#1267

jagged_to_padded_dense supports jagged tensor with inner dense dim is 1 and folded like the following example, but producing the error ``x_offsets.size(), 1 != num_jagged_dim, 0`` in backward.

x_values = [5, 3, 7, 1]
x_offsets = [0, 1, 4]
max_length = 3
->
x_padded = [[5, 0, 0], [3, 7, 1]]

Reviewed By: devashisht

Differential Revision: D39121072

fbshipit-source-id: 02828952851f3382e2281102fba64261bca20882
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D39121072

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 32b6a12.

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

Successfully merging this pull request may close these issues.

None yet

2 participants