Add missing NumPy functions to aesara.tensor

aesara/tensor/__init__.py

@@ -63,13 +63,18 @@ from aesara.tensor.blas import batched_dot, batched_tensordot
 from aesara.tensor.extra_ops import (
     bartlett,
     bincount,
+    broadcast_shape,
+    broadcast_shape_iter,
+    broadcast_to,
     cumprod,
     cumsum,
+    diff,
     fill_diagonal,
     fill_diagonal_offset,
     ravel_multi_index,
     repeat,
     squeeze,
+    unique,
     unravel_index,
 )
 from aesara.tensor.io import *

aesara/tensor/extra_ops.py

@@ -1295,6 +1295,21 @@ class Unique(Op):
             self.axis = None
 
 
+def unique(
+    ar, return_index=False, return_inverse=False, return_counts=False, axis=None
+):
+    """Find the unique elements of an array.
+
+    Returns the sorted unique elements of an array. There are three optional
+    outputs in addition to the unique elements:
+
+    * the indices of the input array that give the unique values
+    * the indices of the unique array that reconstruct the input array
+    * the number of times each unique value comes up in the input array
+    """
+    return Unique(return_index, return_inverse, return_counts, axis)(ar)
+
+
 class UnravelIndex(Op):
     __props__ = ("order",)