text: Title: What is correct behavior of keras.ops.eye?
  Body: Hi, 
  
  I have a question about keras.ops.eye of Keras3.
  
  With Tensorflow, the keras.ops.eye accepts integers and floats.
  However, with torch and jax, only integers are accepted.
  
  Which behavior is correct?
  
  There is no limitation description on documents.
  Because Keras 2 accepts integers and floats, I was confused with it during migration.