1 #include <torch/nn/modules/batchnorm.h>. 2. 3 #include <torch/cuda.h> ... 64 return torch::batch_norm(. 65 input,. 66 weight,. 67 bias,.
確定! 回上一頁