Add transpose (#82)

This commit is contained in:
Fangjun Kuang
2023-03-05 11:27:17 +08:00
committed by GitHub
parent c230cf0a97
commit 422221b3b4
11 changed files with 110 additions and 0 deletions

View File

@@ -0,0 +1,22 @@
// sherpa-onnx/csrc/transpose.h
//
// Copyright (c) 2023 Xiaomi Corporation
#ifndef SHERPA_ONNX_CSRC_TRANSPOSE_H_
#define SHERPA_ONNX_CSRC_TRANSPOSE_H_
#include "onnxruntime_cxx_api.h" // NOLINT
namespace sherpa_onnx {
/** Transpose a 3-D tensor from shape (B, T, C) to (T, B, C).
*
* @param allocator
* @param v A 3-D tensor of shape (B, T, C). Its dataype is T.
*
* @return Return a 3-D tensor of shape (T, B, C). Its datatype is T.
*/
template <typename T = float>
Ort::Value Transpose01(OrtAllocator *allocator, const Ort::Value *v);
} // namespace sherpa_onnx
#endif // SHERPA_ONNX_CSRC_TRANSPOSE_H_