Typedef torch::nn::activation_t

• Defined in File transformerlayer.h

Typedef Documentation

using torch::nn::activation_t = std::variant<enumtype::kReLU, enumtype::kGELU, std::function<Tensor(const Tensor&)>>