{"record":{"id":"89c4946bd2bdbe29","repo":"keras-team/keras","slug":"last-dimension-of-inputs-to-cdist-must-match","errorCode":null,"errorMessage":"Last dimension of inputs to `cdist` must match","messagePattern":"Last dimension of inputs to `cdist` must match","errorType":"validation","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"keras/src/backend/jax/math.py","lineNumber":88,"sourceCode":"    preds_at_label = jnp.where(\n        jnp.isnan(preds_at_label), -jnp.inf, preds_at_label\n    )\n    rank = 1 + jnp.sum(jnp.greater(predictions, preds_at_label), axis=-1)\n    return jnp.less_equal(rank, k)\n\n\ndef logsumexp(x, axis=None, keepdims=False):\n    x = convert_to_tensor(x)\n    return jax.scipy.special.logsumexp(x, axis=axis, keepdims=keepdims)\n\n\ndef cdist(x, y):\n    x = jnp.asarray(x)\n    y = jnp.asarray(y)\n    if x.ndim < 2 or y.ndim < 2:\n        raise ValueError(\"`cdist` inputs must have rank >= 2\")\n    if x.shape[-1] != y.shape[-1]:\n        raise ValueError(\"Last dimension of inputs to `cdist` must match\")\n    diff = jnp.expand_dims(x, -2) - jnp.expand_dims(y, -3)\n    return jnp.sqrt(jnp.sum(diff * diff, axis=-1))\n\n\ndef extract_sequences(x, sequence_length, sequence_stride):\n    *batch_shape, signal_length = x.shape\n    batch_shape = list(batch_shape)\n    x = jnp.reshape(x, (math.prod(batch_shape), signal_length, 1))\n    x = jax.lax.conv_general_dilated_patches(\n        x,\n        (sequence_length,),\n        (sequence_stride,),\n        \"VALID\",\n        dimension_numbers=(\"NTC\", \"OIT\", \"NTC\"),\n    )\n    return jnp.reshape(x, (*batch_shape, *x.shape[-2:]))\n\n","sourceCodeStart":70,"sourceCodeEnd":106,"githubUrl":"https://github.com/keras-team/keras/blob/7a34a03db60bf60042242d6a556fc3be119046a5/keras/src/backend/jax/math.py#L70-L106","documentation":"Error \"Last dimension of inputs to `cdist` must match\" thrown in keras-team/keras.","triggerScenarios":"Thrown at keras/src/backend/jax/math.py:88 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"7a34a03db60bf60042242d6a556fc3be119046a5","analyzedAt":"2026-08-25T21:25:25.994Z","schemaVersion":2},"datasetVersion":"2026-08-26T02:17:13.382Z"}