mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 16:48:03 +08:00
[Functions] Add quantile function (#700)
* Add sort function * Add isfinite function * upgrade isinf isnan * Add Scalar to FDTensor * Add floor, ceil function * add cast functions * Update out_tmp * Update quantile * add gather scatter along axis * finish quantile function * Add quantile unittest * refresh code style for test source code * Add comments * Add full function * Add scalar to fd tensor * Add full unittest * Add functions headers * move fdtensor operators to fastdeploy namespace
This commit is contained in:
@@ -49,5 +49,17 @@ FASTDEPLOY_DECL void Exp(const FDTensor& x, FDTensor* out);
|
||||
*/
|
||||
FASTDEPLOY_DECL void Abs(const FDTensor& x, FDTensor* out);
|
||||
|
||||
/** Computes ceil of x element-wise. Only for float type FDTensor
|
||||
@param x The input tensor.
|
||||
@param out The output tensor which stores the result.
|
||||
*/
|
||||
FASTDEPLOY_DECL void Ceil(const FDTensor& x, FDTensor* out);
|
||||
|
||||
/** Computes floor of x element-wise. Only for float type FDTensor
|
||||
@param x The input tensor.
|
||||
@param out The output tensor which stores the result.
|
||||
*/
|
||||
FASTDEPLOY_DECL void Floor(const FDTensor& x, FDTensor* out);
|
||||
|
||||
} // namespace function
|
||||
} // namespace fastdeploy
|
||||
|
Reference in New Issue
Block a user