class Network
- 类说明
- ~Network() override;
- ActivationLayer *AddActivation(Tensor *input, ActivationKind kind) noexcept;
- ConstantLayer *AddConstant(Dims dimensions, WeightsBuf weights) noexcept;
- ConvolutionLayer *AddConvolution(Tensor *input, int32_t numOutputMap, Dims kernelHW, WeightsBuf weights, WeightsBuf bias) noexcept;
- ElementWiseLayer *AddElementWise(Tensor *input0, Tensor *input1, ElementWiseOperation op) noexcept;
- Tensor *AddInput(const std::string name, DataType type, Dims dim) noexcept;
- MatrixMultiplyLayer *AddMatrixMultiply(Tensor *input0, MatrixOperation type0, Tensor *input1, MatrixOperation type1) noexcept;
- PoolingLayer *AddPooling(Tensor *input, PoolingKind kind, Dims windowSize, Dims outputSize = Dims()) noexcept;
- ReduceLayer *AddReduce(Tensor *input, ReduceOperation operation, std::vector<int64_t> reduceAxes, bool keepDimensions) noexcept;
- ShapeLayer *AddShape(Tensor *input) noexcept;
- ShuffleLayer *AddShuffle(Tensor *input) noexcept;
- ClipLayer *AddClip(Tensor *input, float min, float max) noexcept;
- size_t GetInputNum() const noexcept;
- const std::string GetName() noexcept;
- Tensor* GetOutput(int32_t index) const noexcept;
- size_t GetOutputNum() const noexcept;
- void SetName(const std::string name) noexcept;
- void SetAsOutput(Tensor *tensor) noexcept;