LibTorch之激活函数层

LibTorch之激活函数层

LibTorch之激活函数层

一般使用

::Tensor forward(torch::Tensor x) { // Use one of many tensor manipulation functions. x = torch::relu(fc1->forward(x.reshape({x.size(0), 784}))); x = torch::dropout(x, /*p=*/0.5, /*train=*/is_training()); x = torch::relu(fc2->forward(x)); x = torch::log_softmax(fc3->forward(x), /*dim=*/1); return x; }

Relu激活函数

torch::relu(layer->forward(x))

log_softmax激活函数

torch::log_softmax(layer->forward(x), /*dim=*/1);

LeakyReLU 激活函数

torch::nn::LeakyReLU leakyRelu { nullptr };

x = leakyRelu (x);

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。