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

add transpose const counterpart

parent 3908c88b
......@@ -433,7 +433,7 @@ struct Tensor
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())
{
......@@ -456,6 +456,11 @@ struct Tensor
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 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