And the parameter of torch.nn.BatchNorm2d is the number of dimensions/channels that output from the last layer and come in to the batch norm ...
確定! 回上一頁