complexPyTorch
complexPyTorch copied to clipboard
Fix args for ComplexAvgPool2d
Need to replace dilation and return_indices args with count_include_pad and divisor_override args to be compatible with torch.nn.functional.avg_pool2d