Change the docs of the convolution function
The ranks of input and filter should be N+2 instead of N. Because the function computes N-D convolution, the input's rank should be N+2 with additional batch and input_channel dimensions, and the filter's rank should be N+2 with additional input_channel and output_channel dimensions.
Loading
Please sign in to comment