jax.numpy.ndarray¶

class jax.numpy.ndarray(dtype=None, buffer=None, offset=0, strides=None, order=None)[source]¶
__init__(dtype=None, buffer=None, offset=0, strides=None, order=None)[source]¶

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__([dtype, buffer, offset, strides, order])

Initialize self.

all([axis, out, keepdims])

param axis

any([axis, out, keepdims])

param axis

argmax([axis, out])

param axis

argmin([axis, out])

param axis

argpartition(kth[, axis, kind, order])

argsort([axis, kind, order])

param axis

astype(dtype)

choose(choices[, out, mode])

clip([a_min, a_max, out])

compress(condition[, axis, out])

param axis

conj()

conjugate()

copy()

cumprod([axis, dtype, out])

param axis

cumsum([axis, dtype, out])

param axis

diagonal([offset, axis1, axis2])

param axis1

dot(b, *[, precision])

flatten()

item(*args)

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])

prod([axis, dtype, out, keepdims, initial, …])

param axis

ptp([axis, out, keepdims])

param axis

ravel([order])

repeat(repeats[, axis, total_repeat_length])

param axis

reshape(*args[, order])

round([decimals, out])

searchsorted(v[, side, sorter])

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])

tolist()

trace([offset, axis1, axis2, dtype, out])

param axis1

transpose(*args)

var([axis, dtype, out, ddof, keepdims, where])

param axis

view([dtype, type])

Attributes

at

Helper property for index update functionality.

aval

imag

nbytes

real

weak_type

rtype

bool