* Add pytensor function corresponding to

* Add pytensor function corresponding to

* Cast numpy to node output dtype, rather than depending on config.floatX

Change output type hint back to

* Cast numpy to node output dtype, rather than depending on config.floatX  Change output type hint back to

* Use  rather than  for output equality tests