jax.numpy.ndarray
Warning
This page was created from a pull request (#9655).
jax.numpy.ndarray¶
- class jax.numpy.ndarray(shape, dtype=None, buffer=None, offset=0, strides=None, order=None)[source]¶
-
Methods
__init__(shape[, dtype, buffer, offset, ...])all([axis, out, keepdims])- param axis
any([axis, out, keepdims])- param axis
argmax([axis, out, keepdims])- param axis
argmin([axis, out, keepdims])- param axis
argpartition(kth[, axis, kind, order])- rtype
argsort([axis, kind, order])- param axis
astype(dtype)- rtype
choose(choices[, out, mode])- rtype
clip([a_min, a_max, out])- rtype
compress(condition[, axis, out])- param axis
conj()- rtype
conjugate()- rtype
copy()- rtype
cumprod([axis, dtype, out])- param axis
cumsum([axis, dtype, out])- param axis
diagonal([offset, axis1, axis2])- param axis1
dot(b, *[, precision])- rtype
flatten()- rtype
item(*args)- rtype
max([axis, out, keepdims, initial, where])- param axis
mean([axis, dtype, out, keepdims, where])- param axis
min([axis, out, keepdims, initial, where])- param axis
nonzero(*[, size, fill_value])- rtype
prod([axis, dtype, out, keepdims, initial, ...])- param axis
ptp([axis, out, keepdims])- param axis
ravel([order])- rtype
repeat(repeats[, axis, total_repeat_length])- param axis
reshape(*args[, order])- rtype
round([decimals, out])- rtype
searchsorted(v[, side, sorter])- rtype
sort([axis, kind, order])- param axis
squeeze([axis])- param axis
std([axis, dtype, out, ddof, keepdims, where])- param axis
sum([axis, dtype, out, keepdims, initial, where])- param axis
swapaxes(axis1, axis2)- param axis1
take(indices[, axis, out, mode])- param axis
tobytes([order])- rtype
tolist()- rtype
trace([offset, axis1, axis2, dtype, out])- param axis1
transpose(*args)- rtype
var([axis, dtype, out, ddof, keepdims, where])- param axis
view([dtype, type])- rtype
Attributes
Helper property for index update functionality.
aval- rtype
imag- rtype
nbytes- rtype
real- rtype
weak_type- rtype
dtypendimshapesize