apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 01:22:59 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Model] [Part 1] Refactor PaddleClas module (#508)

Browse Source 
* Split PaddleClas Module refactor

* Split PaddleClas Module refactor

* fix bug
This commit is contained in:
Jason
2022-11-07 15:09:00 +08:00
committed by GitHub
parent 40b099ac99
commit 6633fa3db9
17 changed files with 227 additions and 269 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/core/fd_tensor.h
View File

@@ -93,6 +93,12 @@ struct FASTDEPLOY_DECL FDTensor {
// Total number of elements in this tensor
int Numel() const;
// Get shape of FDTensor
std::vector<int64_t> Shape() const { return shape; }
// Get dtype of FDTensor
FDDataType Dtype() const { return dtype; }
void Resize(size_t nbytes);
void Resize(const std::vector<int64_t>& new_shape);