We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6f72daf commit a54357bCopy full SHA for a54357b
src/array_api_extra/_delegation.py
@@ -117,7 +117,7 @@ def pad(
117
pad_width = xp.asarray(pad_width)
118
pad_width = xp.broadcast_to(pad_width, (x.ndim, 2))
119
pad_width = xp.flip(pad_width, axis=(0,)).flatten()
120
- return xp.nn.functional.pad(x, (pad_width,), value=constant_values)
+ return xp.nn.functional.pad(x, tuple(pad_width), value=constant_values)
121
122
if _delegate(xp, NUMPY, JAX, CUPY):
123
return xp.pad(x, pad_width, mode, constant_values=constant_values)
0 commit comments