add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

* add numeric type hierarchy and issubdtype as well as a set_dtype method to nn.Module with predicate numeric type hierarchy and issubtype is compatible to the [numpy hierarchy](https://github.com/numpy/numpy/blob/220f0ab2c5958f9d49b9a28c69a760657b06d9b9/numpy/_core/numerictypes.py#L42). Closes #285. * nits in docs * unify type category checking * nits in docs * nits in docs * more docs nits * fix callable type --------- Co-authored-by: Awni Hannun <[email protected]>
2024-03-25 12:32:59 -07:00
parent bfb5bad4f0
commit 479051ce1c
26 changed files with 538 additions and 97 deletions
@@ -19,7 +19,6 @@ Array
    array.ndim
    array.shape
    array.size
-    Dtype
    array.abs
    array.all
    array.any
@@ -32,7 +31,6 @@ Array
    array.cumsum
    array.diag
    array.diagonal
-    array.dtype
    array.exp
    array.flatten
    array.log