You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
583 lines
9.4 KiB
Python
583 lines
9.4 KiB
Python
from typing import Any, Literal, TypeAlias, TypeVar, overload
|
|
|
|
import numpy as np
|
|
import numpy.typing as npt
|
|
from numpy import ( # noqa: F401
|
|
False_,
|
|
ScalarType,
|
|
True_,
|
|
__array_namespace_info__,
|
|
__version__,
|
|
abs,
|
|
absolute,
|
|
acos,
|
|
acosh,
|
|
add,
|
|
all,
|
|
allclose,
|
|
amax,
|
|
amin,
|
|
angle,
|
|
any,
|
|
append,
|
|
apply_along_axis,
|
|
apply_over_axes,
|
|
arange,
|
|
arccos,
|
|
arccosh,
|
|
arcsin,
|
|
arcsinh,
|
|
arctan,
|
|
arctan2,
|
|
arctanh,
|
|
argmax,
|
|
argmin,
|
|
argpartition,
|
|
argsort,
|
|
argwhere,
|
|
around,
|
|
array,
|
|
array2string,
|
|
array_equal,
|
|
array_equiv,
|
|
array_repr,
|
|
array_split,
|
|
array_str,
|
|
asanyarray,
|
|
asarray,
|
|
asarray_chkfinite,
|
|
ascontiguousarray,
|
|
asfortranarray,
|
|
asin,
|
|
asinh,
|
|
asmatrix,
|
|
astype,
|
|
atan,
|
|
atan2,
|
|
atanh,
|
|
atleast_1d,
|
|
atleast_2d,
|
|
atleast_3d,
|
|
average,
|
|
bartlett,
|
|
base_repr,
|
|
binary_repr,
|
|
bincount,
|
|
bitwise_and,
|
|
bitwise_count,
|
|
bitwise_invert,
|
|
bitwise_left_shift,
|
|
bitwise_not,
|
|
bitwise_or,
|
|
bitwise_right_shift,
|
|
bitwise_xor,
|
|
blackman,
|
|
block,
|
|
bmat,
|
|
bool,
|
|
bool_,
|
|
broadcast,
|
|
broadcast_arrays,
|
|
broadcast_shapes,
|
|
broadcast_to,
|
|
busday_count,
|
|
busday_offset,
|
|
busdaycalendar,
|
|
byte,
|
|
bytes_,
|
|
c_,
|
|
can_cast,
|
|
cbrt,
|
|
cdouble,
|
|
ceil,
|
|
char,
|
|
character,
|
|
choose,
|
|
clip,
|
|
clongdouble,
|
|
column_stack,
|
|
common_type,
|
|
complex64,
|
|
complex128,
|
|
complex256,
|
|
complexfloating,
|
|
compress,
|
|
concat,
|
|
concatenate,
|
|
conj,
|
|
conjugate,
|
|
convolve,
|
|
copy,
|
|
copysign,
|
|
copyto,
|
|
core,
|
|
corrcoef,
|
|
correlate,
|
|
cos,
|
|
cosh,
|
|
count_nonzero,
|
|
cov,
|
|
cross,
|
|
csingle,
|
|
ctypeslib,
|
|
cumprod,
|
|
cumsum,
|
|
cumulative_prod,
|
|
cumulative_sum,
|
|
datetime64,
|
|
datetime_as_string,
|
|
datetime_data,
|
|
deg2rad,
|
|
degrees,
|
|
delete,
|
|
diag,
|
|
diag_indices,
|
|
diag_indices_from,
|
|
diagflat,
|
|
diagonal,
|
|
diff,
|
|
digitize,
|
|
divide,
|
|
divmod,
|
|
dot,
|
|
double,
|
|
dsplit,
|
|
dstack,
|
|
dtype,
|
|
dtypes,
|
|
e,
|
|
ediff1d,
|
|
einsum,
|
|
einsum_path,
|
|
emath,
|
|
empty_like,
|
|
equal,
|
|
errstate,
|
|
euler_gamma,
|
|
exceptions,
|
|
exp,
|
|
exp2,
|
|
expand_dims,
|
|
expm1,
|
|
extract,
|
|
f2py,
|
|
fabs,
|
|
fft,
|
|
fill_diagonal,
|
|
finfo,
|
|
fix,
|
|
flatiter,
|
|
flatnonzero,
|
|
flexible,
|
|
flip,
|
|
fliplr,
|
|
flipud,
|
|
float16,
|
|
float32,
|
|
float64,
|
|
float128,
|
|
float_power,
|
|
floating,
|
|
floor,
|
|
floor_divide,
|
|
fmax,
|
|
fmin,
|
|
fmod,
|
|
format_float_positional,
|
|
format_float_scientific,
|
|
frexp,
|
|
from_dlpack,
|
|
frombuffer,
|
|
fromfile,
|
|
fromfunction,
|
|
fromiter,
|
|
frompyfunc,
|
|
fromregex,
|
|
fromstring,
|
|
full,
|
|
full_like,
|
|
gcd,
|
|
generic,
|
|
genfromtxt,
|
|
geomspace,
|
|
get_include,
|
|
get_printoptions,
|
|
getbufsize,
|
|
geterr,
|
|
geterrcall,
|
|
gradient,
|
|
greater,
|
|
greater_equal,
|
|
half,
|
|
hamming,
|
|
hanning,
|
|
heaviside,
|
|
histogram,
|
|
histogram2d,
|
|
histogram_bin_edges,
|
|
histogramdd,
|
|
hsplit,
|
|
hstack,
|
|
hypot,
|
|
i0,
|
|
iinfo,
|
|
imag,
|
|
in1d,
|
|
index_exp,
|
|
indices,
|
|
inexact,
|
|
inf,
|
|
info,
|
|
inner,
|
|
insert,
|
|
int8,
|
|
int16,
|
|
int32,
|
|
int64,
|
|
int_,
|
|
intc,
|
|
integer,
|
|
interp,
|
|
intersect1d,
|
|
intp,
|
|
invert,
|
|
is_busday,
|
|
isclose,
|
|
iscomplex,
|
|
iscomplexobj,
|
|
isdtype,
|
|
isfinite,
|
|
isfortran,
|
|
isin,
|
|
isinf,
|
|
isnan,
|
|
isnat,
|
|
isneginf,
|
|
isposinf,
|
|
isreal,
|
|
isrealobj,
|
|
isscalar,
|
|
issubdtype,
|
|
iterable,
|
|
ix_,
|
|
kaiser,
|
|
kron,
|
|
lcm,
|
|
ldexp,
|
|
left_shift,
|
|
less,
|
|
less_equal,
|
|
lexsort,
|
|
lib,
|
|
linalg,
|
|
linspace,
|
|
little_endian,
|
|
load,
|
|
loadtxt,
|
|
log,
|
|
log1p,
|
|
log2,
|
|
log10,
|
|
logaddexp,
|
|
logaddexp2,
|
|
logical_and,
|
|
logical_not,
|
|
logical_or,
|
|
logical_xor,
|
|
logspace,
|
|
long,
|
|
longdouble,
|
|
longlong,
|
|
ma,
|
|
mask_indices,
|
|
matmul,
|
|
matrix,
|
|
matrix_transpose,
|
|
matvec,
|
|
max,
|
|
maximum,
|
|
may_share_memory,
|
|
mean,
|
|
median,
|
|
memmap,
|
|
meshgrid,
|
|
mgrid,
|
|
min,
|
|
min_scalar_type,
|
|
minimum,
|
|
mintypecode,
|
|
mod,
|
|
modf,
|
|
moveaxis,
|
|
multiply,
|
|
nan,
|
|
nan_to_num,
|
|
nanargmax,
|
|
nanargmin,
|
|
nancumprod,
|
|
nancumsum,
|
|
nanmax,
|
|
nanmean,
|
|
nanmedian,
|
|
nanmin,
|
|
nanpercentile,
|
|
nanprod,
|
|
nanquantile,
|
|
nanstd,
|
|
nansum,
|
|
nanvar,
|
|
ndarray,
|
|
ndenumerate,
|
|
ndim,
|
|
ndindex,
|
|
nditer,
|
|
negative,
|
|
nested_iters,
|
|
newaxis,
|
|
nextafter,
|
|
nonzero,
|
|
not_equal,
|
|
number,
|
|
object_,
|
|
ogrid,
|
|
ones_like,
|
|
outer,
|
|
packbits,
|
|
pad,
|
|
partition,
|
|
percentile,
|
|
permute_dims,
|
|
pi,
|
|
piecewise,
|
|
place,
|
|
poly,
|
|
poly1d,
|
|
polyadd,
|
|
polyder,
|
|
polydiv,
|
|
polyfit,
|
|
polyint,
|
|
polymul,
|
|
polynomial,
|
|
polysub,
|
|
polyval,
|
|
positive,
|
|
pow,
|
|
power,
|
|
printoptions,
|
|
prod,
|
|
promote_types,
|
|
ptp,
|
|
put,
|
|
put_along_axis,
|
|
putmask,
|
|
quantile,
|
|
r_,
|
|
rad2deg,
|
|
radians,
|
|
random,
|
|
ravel,
|
|
ravel_multi_index,
|
|
real,
|
|
real_if_close,
|
|
rec,
|
|
recarray,
|
|
reciprocal,
|
|
record,
|
|
remainder,
|
|
repeat,
|
|
require,
|
|
reshape,
|
|
resize,
|
|
result_type,
|
|
right_shift,
|
|
rint,
|
|
roll,
|
|
rollaxis,
|
|
roots,
|
|
rot90,
|
|
round,
|
|
row_stack,
|
|
s_,
|
|
save,
|
|
savetxt,
|
|
savez,
|
|
savez_compressed,
|
|
sctypeDict,
|
|
searchsorted,
|
|
select,
|
|
set_printoptions,
|
|
setbufsize,
|
|
setdiff1d,
|
|
seterr,
|
|
seterrcall,
|
|
setxor1d,
|
|
shape,
|
|
shares_memory,
|
|
short,
|
|
show_config,
|
|
show_runtime,
|
|
sign,
|
|
signbit,
|
|
signedinteger,
|
|
sin,
|
|
sinc,
|
|
single,
|
|
sinh,
|
|
size,
|
|
sort,
|
|
sort_complex,
|
|
spacing,
|
|
split,
|
|
sqrt,
|
|
square,
|
|
squeeze,
|
|
stack,
|
|
std,
|
|
str_,
|
|
strings,
|
|
subtract,
|
|
sum,
|
|
swapaxes,
|
|
take,
|
|
take_along_axis,
|
|
tan,
|
|
tanh,
|
|
tensordot,
|
|
test,
|
|
testing,
|
|
tile,
|
|
timedelta64,
|
|
trace,
|
|
transpose,
|
|
trapezoid,
|
|
trapz,
|
|
tri,
|
|
tril,
|
|
tril_indices,
|
|
tril_indices_from,
|
|
trim_zeros,
|
|
triu,
|
|
triu_indices,
|
|
triu_indices_from,
|
|
true_divide,
|
|
trunc,
|
|
typecodes,
|
|
typename,
|
|
typing,
|
|
ubyte,
|
|
ufunc,
|
|
uint,
|
|
uint8,
|
|
uint16,
|
|
uint32,
|
|
uint64,
|
|
uintc,
|
|
uintp,
|
|
ulong,
|
|
ulonglong,
|
|
union1d,
|
|
unique,
|
|
unique_all,
|
|
unique_counts,
|
|
unique_inverse,
|
|
unique_values,
|
|
unpackbits,
|
|
unravel_index,
|
|
unsignedinteger,
|
|
unstack,
|
|
unwrap,
|
|
ushort,
|
|
vander,
|
|
var,
|
|
vdot,
|
|
vecdot,
|
|
vecmat,
|
|
vectorize,
|
|
void,
|
|
vsplit,
|
|
vstack,
|
|
where,
|
|
zeros_like,
|
|
)
|
|
from numpy._typing import _ArrayLike, _DTypeLike
|
|
|
|
__all__ = ["rand", "randn", "repmat"]
|
|
__all__ += np.__all__
|
|
|
|
###
|
|
|
|
_T = TypeVar("_T", bound=np.generic)
|
|
_Matrix: TypeAlias = np.matrix[tuple[int, int], np.dtype[_T]]
|
|
_Order: TypeAlias = Literal["C", "F"]
|
|
|
|
###
|
|
|
|
#
|
|
@overload
|
|
def empty(shape: int | tuple[int, int], dtype: None = None, order: _Order = "C") -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def empty(shape: int | tuple[int, int], dtype: _DTypeLike[_T], order: _Order = "C") -> _Matrix[_T]: ...
|
|
@overload
|
|
def empty(shape: int | tuple[int, int], dtype: npt.DTypeLike, order: _Order = "C") -> _Matrix[Any]: ...
|
|
|
|
#
|
|
@overload
|
|
def ones(shape: int | tuple[int, int], dtype: None = None, order: _Order = "C") -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def ones(shape: int | tuple[int, int], dtype: _DTypeLike[_T], order: _Order = "C") -> _Matrix[_T]: ...
|
|
@overload
|
|
def ones(shape: int | tuple[int, int], dtype: npt.DTypeLike, order: _Order = "C") -> _Matrix[Any]: ...
|
|
|
|
#
|
|
@overload
|
|
def zeros(shape: int | tuple[int, int], dtype: None = None, order: _Order = "C") -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def zeros(shape: int | tuple[int, int], dtype: _DTypeLike[_T], order: _Order = "C") -> _Matrix[_T]: ...
|
|
@overload
|
|
def zeros(shape: int | tuple[int, int], dtype: npt.DTypeLike, order: _Order = "C") -> _Matrix[Any]: ...
|
|
|
|
#
|
|
@overload
|
|
def identity(n: int, dtype: None = None) -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def identity(n: int, dtype: _DTypeLike[_T]) -> _Matrix[_T]: ...
|
|
@overload
|
|
def identity(n: int, dtype: npt.DTypeLike | None = None) -> _Matrix[Any]: ...
|
|
|
|
#
|
|
@overload
|
|
def eye(
|
|
n: int,
|
|
M: int | None = None,
|
|
k: int = 0,
|
|
dtype: type[np.float64] | None = ...,
|
|
order: _Order = "C",
|
|
) -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def eye(n: int, M: int | None, k: int, dtype: _DTypeLike[_T], order: _Order = "C") -> _Matrix[_T]: ...
|
|
@overload
|
|
def eye(n: int, M: int | None = None, k: int = 0, *, dtype: _DTypeLike[_T], order: _Order = "C") -> _Matrix[_T]: ...
|
|
@overload
|
|
def eye(n: int, M: int | None = None, k: int = 0, dtype: npt.DTypeLike = ..., order: _Order = "C") -> _Matrix[Any]: ...
|
|
|
|
#
|
|
@overload
|
|
def rand(arg: int | tuple[()] | tuple[int] | tuple[int, int], /) -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def rand(arg: int, /, *args: int) -> _Matrix[np.float64]: ...
|
|
|
|
#
|
|
@overload
|
|
def randn(arg: int | tuple[()] | tuple[int] | tuple[int, int], /) -> _Matrix[np.float64]: ...
|
|
@overload
|
|
def randn(arg: int, /, *args: int) -> _Matrix[np.float64]: ...
|
|
|
|
#
|
|
@overload
|
|
def repmat(a: _Matrix[_T], m: int, n: int) -> _Matrix[_T]: ...
|
|
@overload
|
|
def repmat(a: _ArrayLike[_T], m: int, n: int) -> npt.NDArray[_T]: ...
|
|
@overload
|
|
def repmat(a: npt.ArrayLike, m: int, n: int) -> npt.NDArray[Any]: ...
|