Commit 4f6d52c1 authored by Anthony Chang's avatar Anthony Chang
Browse files

add transpose const counterpart

parent 3908c88b
...@@ -433,7 +433,7 @@ struct Tensor ...@@ -433,7 +433,7 @@ struct Tensor
return mData[mDesc.GetOffsetFromMultiIndex(idx)]; return mData[mDesc.GetOffsetFromMultiIndex(idx)];
} }
Tensor<T> Transpose(std::vector<size_t> axes = {}) Tensor<T> Transpose(std::vector<size_t> axes = {}) const
{ {
if(axes.empty()) if(axes.empty())
{ {
...@@ -456,6 +456,11 @@ struct Tensor ...@@ -456,6 +456,11 @@ struct Tensor
return ret; return ret;
} }
Tensor<T> Transpose(std::vector<size_t> axes = {})
{
return const_cast<Tensor<T> const*>(this)->Transpose(axes);
}
typename Data::iterator begin() { return mData.begin(); } typename Data::iterator begin() { return mData.begin(); }
typename Data::iterator end() { return mData.end(); } typename Data::iterator end() { return mData.end(); }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment