1 Star 0 Fork 0

爱丽丝妹妹与数学妖精/TensorLab

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
layer 892 Bytes
一键复制 编辑 原始数据 按行查看 历史
AuroraKarow 提交于 2022-05-13 00:27 . Decouple layer structure
#pragma once
#include "netbatlib"
#include "async"
#define LAYER_BEGIN namespace layer {
#define LAYER_END }
#define _CONV conv::
#define _FC fc::
#define _ADA ada::
// NULL = no activation function
#define SIGMOID 1
#define RELU 2
#define SOFTMAX 3
#define POOL_AVG 0
#define POOL_MAX 1
#define POOL_GAG 2
#define ACT 0
#define ACT_VT 1
#define ACT_FT 2
#define FC 3
#define CONV 4
#define POOL 5
#define FC_BN 6
#define CONV_BN 7
#define TRANS 8
#define CONV_IM2COL 9
#define CONV_BN_IM2COL 10
#define POOL_IM2COL 11
#define TRANS_IM2COL 12
#define PAD_CROP_IM2COL 13
#include "layer.hpp"
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/AliceSisMathElf/TensorLab.git
[email protected]:AliceSisMathElf/TensorLab.git
AliceSisMathElf
TensorLab
TensorLab
master

搜索帮助