I do not understand why are some Function
class splited into the thnn
directory in torch/nn/_functions
. For example, there are two activation.py
in torch/nn/_functions/
and torch/nn/_functions/thnn/
.
3 Likes
I believe thnn is for functions directly implemented in Torch C or Cuda.
1 Like
th means torch, nn means neuron network.
4 Likes