Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
xerus
xerus
Commits
168b2ff9
Commit
168b2ff9
authored
Apr 09, 2020
by
Philipp Trunschke
Browse files
bugfix in Tensor.from_buffer
parent
81952ee5
Changes
1
Hide whitespace changes
Inline
Side-by-side
src/xerus/python/tensor.cpp
View file @
168b2ff9
...
...
@@ -5,9 +5,10 @@ std::vector<size_t> strides_from_dimensions_and_item_size(const std::vector<size
const
size_t
ndim
=
_dimensions
.
size
();
std
::
vector
<
size_t
>
strides
(
ndim
,
0
);
if
(
ndim
>
0
)
{
strides
[
0
]
=
_item_size
;
strides
[
ndim
-
1
]
=
_item_size
;
for
(
size_t
i
=
0
;
i
<
ndim
-
1
;
++
i
)
{
strides
[
i
+
1
]
=
strides
[
i
]
*
_dimensions
[
i
];
size_t
rev_i
=
ndim
-
1
-
i
;
strides
[
rev_i
-
1
]
=
_dimensions
[
rev_i
]
*
strides
[
rev_i
];
}
}
return
strides
;
...
...
@@ -81,7 +82,8 @@ void expose_tensor(module& m) {
}
Tensor
result
(
dims
,
Tensor
::
Representation
::
Dense
,
Tensor
::
Initialisation
::
None
);
*
(
result
.
denseData
)
=
info
.
ptr
;
/* *(result.override_dense_data()) = static_cast<double*>(info.ptr); */
misc
::
copy
(
result
.
get_unsanitized_dense_data
(),
static_cast
<
double
*>
(
info
.
ptr
),
result
.
size
);
return
result
;
})
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment