LibTorch之激活函数层

LibTorch之激活函数层

一般使用

 torch::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);