jax.numpy.ndarray#

class jax.numpy.ndarray(shape, dtype=None, buffer=None, offset=0, strides=None, order=None)[source]#
__init__(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

Any

argsort([axis, kind, order])

param axis

astype(dtype)

rtype

Any

choose(choices[, out, mode])

rtype

Any

clip([a_min, a_max, out])

rtype

Any

compress(condition[, axis, out])

param axis

conj()

rtype

Any

conjugate()

rtype

Any

copy()

rtype

Any

cumprod([axis, dtype, out])

param axis

cumsum([axis, dtype, out])

param axis

diagonal([offset, axis1, axis2])

param axis1

dot(b, *[, precision])

rtype

Any

flatten()

rtype

Any

item(*args)

rtype

Any

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

Any

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

param axis

ptp([axis, out, keepdims])

param axis

ravel([order])

rtype

Any

repeat(repeats[, axis, total_repeat_length])

param axis

reshape(*args[, order])

rtype

Any

round([decimals, out])

rtype

Any

searchsorted(v[, side, sorter])

rtype

Any

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

Any

tolist()

rtype

Any

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

param axis1

transpose(*args)

rtype

Any

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

param axis

view([dtype, type])

rtype

Any

Attributes

at

Helper property for index update functionality.

aval

rtype

Any

imag

rtype

Any

nbytes

rtype

Any

real

rtype

Any

weak_type

rtype

bool

dtype

ndim

shape

size