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 915a368 commit 05dac9eCopy full SHA for 05dac9e
pytensor/xtensor/shape.py
@@ -360,7 +360,7 @@ def squeeze(x, dim=None):
360
The input tensor
361
dim : str or None or iterable of str, optional
362
The name(s) of the dimension(s) to remove. If None, all dimensions of size 1
363
- (known statically) will be removed. Dimensions with symbolic shape will be retained.
+ (known statically) will be removed. Dimensions with unknown static shape will be retained, even if they have size 1 at runtime.
364
365
Returns
366
-------
0 commit comments