mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
27 lines
488 B
C
27 lines
488 B
C
|
#ifndef LAYER_TENSORSLICE_H
|
||
|
|
#define LAYER_TENSORSLICE_H
|
||
|
|
|
||
|
|
#include "layer.h"
|
||
|
|
|
||
|
|
namespace mmlab {
|
||
|
|
|
||
|
|
class TensorSlice : public ncnn::Layer {
|
||
|
|
public:
|
||
|
|
TensorSlice();
|
||
|
|
|
||
|
|
virtual int load_param(const ncnn::ParamDict& pd);
|
||
|
|
|
||
|
|
virtual int forward(const ncnn::Mat& bottom_blob, ncnn::Mat& top_blob,
|
||
|
|
const ncnn::Option& opt) const;
|
||
|
|
|
||
|
|
public:
|
||
|
|
ncnn::Mat starts;
|
||
|
|
ncnn::Mat ends;
|
||
|
|
ncnn::Mat axes;
|
||
|
|
ncnn::Mat steps;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace mmlab
|
||
|
|
|
||
|
|
#endif // LAYER_TENSORSLICE_H
|