Add a GPUArray.squeeze() method.
Sometimes useful in cases where numpy and GPUArray shapes don't match due to length 1 dimensions. This can occur when slicing GPU and CPU arrays and helps to suppress warnings about non-matching array shapes.
Loading
Please register or sign in to comment