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.

Sign up for GitHub

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

Jump to bottom

torch.ByteTensor and nonzero() method #538

Closed
SergeyMalashenko opened this issue Oct 11, 2016 · 5 comments
Closed

torch.ByteTensor and nonzero() method #538

SergeyMalashenko opened this issue Oct 11, 2016 · 5 comments

Comments

@SergeyMalashenko
Copy link

Can I apply nonzero() method to torch.ByteTensor object? or do I need to convert to torch.FloatTensor at first?
@fmassa
Copy link
Contributor

fmassa commented Oct 11, 2016

Yes, you can apply nonzero to torch.ByteTensors.

@SergeyMalashenko
Copy link
Author

Ok. Can I apply the same method to torch.CudaByteTensor?

@fmassa
Copy link
Contributor

fmassa commented Oct 11, 2016

There is unfortunately no nonzero() implementation for CUDA yet (just realized the issue was opened in the cutorch repo)

@JoostvDoorn
Copy link
Contributor

Some time ago @soumith mentioned that Facebook implemented nonzero for CUDA in #70

@soumith
Copy link
Member

soumith commented Nov 1, 2016

this is now pushed into master

@soumith soumith closed this as completed Nov 1, 2016
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@soumith @JoostvDoorn @fmassa @SergeyMalashenko