Add Transpose function (#91)

* Add Transpose function

* csrcs->csrc

* Add transpose unittest

* Add reduce_max_large_dim unittest
This commit is contained in:
Jack Zhou
2022-08-10 19:00:16 +08:00
committed by GitHub
parent bf5affb510
commit 7fb8dd7916
9 changed files with 298 additions and 13 deletions

View File

@@ -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