-
Notifications
You must be signed in to change notification settings - Fork 230
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
KnetArray transpose and permutedims missing. #39
Comments
Is permutedims for 4D and 5D practical? Necessary? |
I am leaving this issue open for the general N-D implementation. |
@ilkerkesen do we actually have a general N-D implementation (beyond 5-D) that works? If so, let's close this. |
Back then I implemented a general permutedims operation, but couldn't port
it to Knet. Yes, we have, but it needs to be ported.
On Dec 24, 2017 6:55 PM, "denizyuret" <notifications@github.com> wrote:
@ilkerkesen <https://github.com/ilkerkesen> do we actually have a general
N-D implementation (beyond 5-D) that works? If so, let's close this.
—
You are receiving this because you were mentioned.
Reply to this email directly, view it on GitHub
<#39 (comment)>,
or mute the thread
<https://github.com/notifications/unsubscribe-auth/ACrjQ4p-NA4lyk5yCrb3QOlRo0rwbegwks5tDnP_gaJpZM4KwX7U>
.
|
So far so good,
I've looked at PyTorch's source code and I think this is the part where transpose magic happens. As different from Julia, PyTorch doesn't have a
As a first step, I'm going to implement a |
Caffe2 has a multi-dimensional transpose operation in the way we desire. It takes advantage of cudnn's |
Solved using CuArrays under permutedims branch, please test. |
No description provided.
The text was updated successfully, but these errors were encountered: