在Pytorch中,BN的調用可能是這樣的. torch.nn.BatchNorm2d -> F.batch_norm -> C++ -> yaml -> C++. 和yaml相關的文件, 該文件中BN可以根據設備選擇不同的實現方式 ...
確定! 回上一頁