mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-17 22:21:48 +08:00
Add Transpose function (#91)
* Add Transpose function * csrcs->csrc * Add transpose unittest * Add reduce_max_large_dim unittest
This commit is contained in:
@@ -46,4 +46,13 @@ bool ReadBinaryFromFile(const std::string& file, std::string* contents) {
|
||||
return true;
|
||||
}
|
||||
|
||||
std::vector<int64_t> GetStride(const std::vector<int64_t>& dims) {
|
||||
auto dims_size = dims.size();
|
||||
std::vector<int64_t> result(dims_size, 1);
|
||||
for (int i = dims_size - 2; i >= 0; --i) {
|
||||
result[i] = result[i + 1] * dims[i + 1];
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
} // namespace fastdeploy
|
||||
|
Reference in New Issue
Block a user