module Torchrb::NN