Could you check your code for any usage of squeeze()?
If you don’t pass a dim argument to this method, all dimensions with a size of 1 will be removed.
I don’t know how you are using pysift, but could the batch size be reduced to 1, which would then remove this dimension via squeeze()?