deepmd._vendors.ndtensorflow._namespace

Contents

deepmd._vendors.ndtensorflow._namespace#

Attributes#

Classes#

Array

User-facing TensorFlow-backed array object.

UniqueAllResult

UniqueCountsResult

UniqueInverseResult

Functions#

result_type(→ DType)

asarray(→ deepmd._vendors.ndtensorflow._array.Array)

astype(→ deepmd._vendors.ndtensorflow._array.Array)

from_dlpack(→ deepmd._vendors.ndtensorflow._array.Array)

arange(→ deepmd._vendors.ndtensorflow._array.Array)

empty(→ deepmd._vendors.ndtensorflow._array.Array)

empty_like(→ deepmd._vendors.ndtensorflow._array.Array)

eye(→ deepmd._vendors.ndtensorflow._array.Array)

full(→ deepmd._vendors.ndtensorflow._array.Array)

full_like(→ deepmd._vendors.ndtensorflow._array.Array)

linspace(→ deepmd._vendors.ndtensorflow._array.Array)

ones(→ deepmd._vendors.ndtensorflow._array.Array)

ones_like(→ deepmd._vendors.ndtensorflow._array.Array)

zeros(→ deepmd._vendors.ndtensorflow._array.Array)

zeros_like(→ deepmd._vendors.ndtensorflow._array.Array)

tril(→ deepmd._vendors.ndtensorflow._array.Array)

triu(→ deepmd._vendors.ndtensorflow._array.Array)

can_cast(→ bool)

isdtype(→ bool)

finfo(→ _FInfo)

iinfo(→ _IInfo)

abs(→ deepmd._vendors.ndtensorflow._array.Array)

expm1(→ deepmd._vendors.ndtensorflow._array.Array)

tanh(→ deepmd._vendors.ndtensorflow._array.Array)

remainder(→ deepmd._vendors.ndtensorflow._array.Array)

bitwise_and(→ deepmd._vendors.ndtensorflow._array.Array)

bitwise_left_shift(...)

bitwise_invert(→ deepmd._vendors.ndtensorflow._array.Array)

bitwise_or(→ deepmd._vendors.ndtensorflow._array.Array)

bitwise_right_shift(...)

bitwise_xor(→ deepmd._vendors.ndtensorflow._array.Array)

ceil(→ deepmd._vendors.ndtensorflow._array.Array)

floor(→ deepmd._vendors.ndtensorflow._array.Array)

trunc(→ deepmd._vendors.ndtensorflow._array.Array)

copysign(→ deepmd._vendors.ndtensorflow._array.Array)

hypot(→ deepmd._vendors.ndtensorflow._array.Array)

imag(→ deepmd._vendors.ndtensorflow._array.Array)

isfinite(→ deepmd._vendors.ndtensorflow._array.Array)

isinf(→ deepmd._vendors.ndtensorflow._array.Array)

isnan(→ deepmd._vendors.ndtensorflow._array.Array)

log(→ deepmd._vendors.ndtensorflow._array.Array)

floor_divide(→ deepmd._vendors.ndtensorflow._array.Array)

log1p(→ deepmd._vendors.ndtensorflow._array.Array)

log2(→ deepmd._vendors.ndtensorflow._array.Array)

log10(→ deepmd._vendors.ndtensorflow._array.Array)

negative(→ deepmd._vendors.ndtensorflow._array.Array)

pow(→ deepmd._vendors.ndtensorflow._array.Array)

logaddexp(→ deepmd._vendors.ndtensorflow._array.Array)

nextafter(→ deepmd._vendors.ndtensorflow._array.Array)

real(→ deepmd._vendors.ndtensorflow._array.Array)

reciprocal(→ deepmd._vendors.ndtensorflow._array.Array)

round(→ deepmd._vendors.ndtensorflow._array.Array)

sign(→ deepmd._vendors.ndtensorflow._array.Array)

signbit(→ deepmd._vendors.ndtensorflow._array.Array)

clip(→ deepmd._vendors.ndtensorflow._array.Array)

all(→ deepmd._vendors.ndtensorflow._array.Array)

any(→ deepmd._vendors.ndtensorflow._array.Array)

max(→ deepmd._vendors.ndtensorflow._array.Array)

min(→ deepmd._vendors.ndtensorflow._array.Array)

mean(→ deepmd._vendors.ndtensorflow._array.Array)

prod(→ deepmd._vendors.ndtensorflow._array.Array)

sum(→ deepmd._vendors.ndtensorflow._array.Array)

var(→ deepmd._vendors.ndtensorflow._array.Array)

std(→ deepmd._vendors.ndtensorflow._array.Array)

cumulative_sum(→ deepmd._vendors.ndtensorflow._array.Array)

cumulative_prod(...)

diff(→ deepmd._vendors.ndtensorflow._array.Array)

argsort(→ deepmd._vendors.ndtensorflow._array.Array)

sort(→ deepmd._vendors.ndtensorflow._array.Array)

take(→ deepmd._vendors.ndtensorflow._array.Array)

take_along_axis(...)

matmul(→ deepmd._vendors.ndtensorflow._array.Array)

matrix_transpose(...)

tensordot(→ deepmd._vendors.ndtensorflow._array.Array)

vecdot(→ deepmd._vendors.ndtensorflow._array.Array)

broadcast_shapes(→ tuple[int, ...])

broadcast_to(→ deepmd._vendors.ndtensorflow._array.Array)

broadcast_arrays(...)

concat(→ deepmd._vendors.ndtensorflow._array.Array)

expand_dims(→ deepmd._vendors.ndtensorflow._array.Array)

flip(→ deepmd._vendors.ndtensorflow._array.Array)

meshgrid(...)

moveaxis(→ deepmd._vendors.ndtensorflow._array.Array)

permute_dims(→ deepmd._vendors.ndtensorflow._array.Array)

transpose(→ deepmd._vendors.ndtensorflow._array.Array)

einsum(→ deepmd._vendors.ndtensorflow._array.Array)

repeat(→ deepmd._vendors.ndtensorflow._array.Array)

reshape(→ deepmd._vendors.ndtensorflow._array.Array)

roll(→ deepmd._vendors.ndtensorflow._array.Array)

squeeze(→ deepmd._vendors.ndtensorflow._array.Array)

stack(→ deepmd._vendors.ndtensorflow._array.Array)

tile(→ deepmd._vendors.ndtensorflow._array.Array)

unstack(...)

argmax(→ deepmd._vendors.ndtensorflow._array.Array)

argmin(→ deepmd._vendors.ndtensorflow._array.Array)

count_nonzero(→ deepmd._vendors.ndtensorflow._array.Array)

nonzero(...)

searchsorted(→ deepmd._vendors.ndtensorflow._array.Array)

where(→ deepmd._vendors.ndtensorflow._array.Array)

unique_all(→ UniqueAllResult)

unique_counts(→ UniqueCountsResult)

unique_inverse(→ UniqueInverseResult)

unique_values(→ deepmd._vendors.ndtensorflow._array.Array)

isin(→ deepmd._vendors.ndtensorflow._array.Array)

Module Contents#

class deepmd._vendors.ndtensorflow._namespace.Array(tensor: Any | None = None, /)[source]#

Bases: tf.experimental.ExtensionType

User-facing TensorFlow-backed array object.

The object owns the Array API surface. TensorFlow tensors remain plain TensorFlow tensors; no TensorFlow class is patched.

__array_priority__ = 1[source]#
__composite_gradient__[source]#
_tensor: tf.Tensor[source]#
classmethod _from_tensor(tensor: Any, /) → Array[source]#
_replace_tensor(tensor: tf.Tensor) → None[source]#
unwrap() → tf.Tensor[source]#

Return the wrapped TensorFlow tensor.

__tf_tensor__(dtype: tensorflow.DType | None = None, name: str | None = None) → tf.Tensor[source]#
property device: str[source]#
property dtype: tensorflow.DType[source]#
property mT: Array[source]#
property ndim: int[source]#
property shape: tuple[int | tf.Tensor, ...][source]#
property size: int | tf.Tensor[source]#
property T: Array[source]#
astype(dtype: tensorflow.DType, /, *, copy: bool = True, device: str | None = None) → Array[source]#
reshape(*shape: Any, copy: bool | None = None) → Array[source]#
ravel() → Array[source]#
squeeze(axis: int | tuple[int, ...] | None = None) → Array[source]#
to_device(device: str, /, *, stream: int | Any | None = None) → Array[source]#
__dlpack__(*, stream: int | Any | None = None, max_version: tuple[int, int] | None = None, dl_device: tuple[int, int] | None = None, copy: bool | None = None) → Any[source]#
__dlpack_device__() → tuple[DLDeviceType, int][source]#
__array_namespace__(/, *, api_version: str | None = None) → Any[source]#
__array__(dtype: Any | None = None) → Any[source]#
__len__() → int[source]#
__iter__() → collections.abc.Iterator[Array][source]#
__getitem__(key: Any, /) → Array[source]#
__setitem__(key: Any, value: Any, /) → None[source]#
_scalar_value() → Any[source]#
__bool__(/) → bool[source]#
__complex__(/) → complex[source]#
__float__(/) → float[source]#
__index__(/) → int[source]#
__int__(/) → int[source]#
__repr__() → str[source]#
__eq__(other: Any) → Array[source]#
__ne__(other: Any) → Array[source]#
__abs__() → Array[source]#
__invert__() → Array[source]#
__neg__() → Array[source]#
__pos__() → Array[source]#
deepmd._vendors.ndtensorflow._namespace.DType[source]#
deepmd._vendors.ndtensorflow._namespace.Device[source]#
deepmd._vendors.ndtensorflow._namespace.bool[source]#
deepmd._vendors.ndtensorflow._namespace.int8[source]#
deepmd._vendors.ndtensorflow._namespace.int16[source]#
deepmd._vendors.ndtensorflow._namespace.int32[source]#
deepmd._vendors.ndtensorflow._namespace.int64[source]#
deepmd._vendors.ndtensorflow._namespace.uint8[source]#
deepmd._vendors.ndtensorflow._namespace.uint16[source]#
deepmd._vendors.ndtensorflow._namespace.uint32[source]#
deepmd._vendors.ndtensorflow._namespace.uint64[source]#
deepmd._vendors.ndtensorflow._namespace.float16[source]#
deepmd._vendors.ndtensorflow._namespace.bfloat16[source]#
deepmd._vendors.ndtensorflow._namespace.float32[source]#
deepmd._vendors.ndtensorflow._namespace.float64[source]#
deepmd._vendors.ndtensorflow._namespace.complex64[source]#
deepmd._vendors.ndtensorflow._namespace.complex128[source]#
deepmd._vendors.ndtensorflow._namespace.newaxis = None[source]#
deepmd._vendors.ndtensorflow._namespace.e = 2.718281828459045[source]#
deepmd._vendors.ndtensorflow._namespace.inf = inf[source]#
deepmd._vendors.ndtensorflow._namespace.nan[source]#
deepmd._vendors.ndtensorflow._namespace.pi = 3.141592653589793[source]#
class deepmd._vendors.ndtensorflow._namespace.UniqueAllResult[source]#

Bases: tuple

values[source]#
indices[source]#
inverse_indices[source]#
counts[source]#
class deepmd._vendors.ndtensorflow._namespace.UniqueCountsResult[source]#

Bases: tuple

values[source]#
counts[source]#
class deepmd._vendors.ndtensorflow._namespace.UniqueInverseResult[source]#

Bases: tuple

values[source]#
inverse_indices[source]#
deepmd._vendors.ndtensorflow._namespace.result_type(*arrays_and_dtypes: deepmd._vendors.ndtensorflow._array.Array | tf.Tensor | DType | complex) → DType[source]#
deepmd._vendors.ndtensorflow._namespace.asarray(obj: Any, /, *, dtype: DType | None = None, device: Device | None = None, copy: bool | None = None, **kwargs: Any) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.astype(x: deepmd._vendors.ndtensorflow._array.Array, dtype: DType, /, *, copy: bool = True, device: Device | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.from_dlpack(x: Any, /, *, device: Device | None = None, copy: bool | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.arange(start: float, /, stop: float | None = None, step: float = 1, *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.empty(shape: int | tuple[int, ...], *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.empty_like(x: deepmd._vendors.ndtensorflow._array.Array, /, *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.eye(n_rows: int, n_cols: int | None = None, /, *, k: int = 0, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.full(shape: int | tuple[int, ...], fill_value: complex, *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.full_like(x: deepmd._vendors.ndtensorflow._array.Array, /, fill_value: complex, *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.linspace(start: float, stop: float, /, num: int, *, dtype: DType | None = None, device: Device | None = None, endpoint: bool = True, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.ones(shape: int | tuple[int, ...], *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.ones_like(x: deepmd._vendors.ndtensorflow._array.Array, /, *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.zeros(shape: int | tuple[int, ...], *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.zeros_like(x: deepmd._vendors.ndtensorflow._array.Array, /, *, dtype: DType | None = None, device: Device | None = None, **kwargs: object) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.tril(x: deepmd._vendors.ndtensorflow._array.Array, /, *, k: int = 0) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.triu(x: deepmd._vendors.ndtensorflow._array.Array, /, *, k: int = 0) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.can_cast(from_: DType | deepmd._vendors.ndtensorflow._array.Array, to: DType, /) → bool[source]#
deepmd._vendors.ndtensorflow._namespace.isdtype(dtype: DType, kind: DType | str | tuple[DType | str, ...], *, _tuple: bool = True) → bool[source]#
deepmd._vendors.ndtensorflow._namespace.finfo(type_: DType | deepmd._vendors.ndtensorflow._array.Array, /) → _FInfo[source]#
deepmd._vendors.ndtensorflow._namespace.iinfo(type_: DType | deepmd._vendors.ndtensorflow._array.Array, /) → _IInfo[source]#
deepmd._vendors.ndtensorflow._namespace.abs(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.acos[source]#
deepmd._vendors.ndtensorflow._namespace.acosh[source]#
deepmd._vendors.ndtensorflow._namespace.asin[source]#
deepmd._vendors.ndtensorflow._namespace.asinh[source]#
deepmd._vendors.ndtensorflow._namespace.atan[source]#
deepmd._vendors.ndtensorflow._namespace.atan2[source]#
deepmd._vendors.ndtensorflow._namespace.atanh[source]#
deepmd._vendors.ndtensorflow._namespace.add[source]#
deepmd._vendors.ndtensorflow._namespace.conj[source]#
deepmd._vendors.ndtensorflow._namespace.cos[source]#
deepmd._vendors.ndtensorflow._namespace.cosh[source]#
deepmd._vendors.ndtensorflow._namespace.divide[source]#
deepmd._vendors.ndtensorflow._namespace.equal[source]#
deepmd._vendors.ndtensorflow._namespace.exp[source]#
deepmd._vendors.ndtensorflow._namespace.greater[source]#
deepmd._vendors.ndtensorflow._namespace.greater_equal[source]#
deepmd._vendors.ndtensorflow._namespace.less[source]#
deepmd._vendors.ndtensorflow._namespace.less_equal[source]#
deepmd._vendors.ndtensorflow._namespace.logical_and[source]#
deepmd._vendors.ndtensorflow._namespace.logical_not[source]#
deepmd._vendors.ndtensorflow._namespace.logical_or[source]#
deepmd._vendors.ndtensorflow._namespace.logical_xor[source]#
deepmd._vendors.ndtensorflow._namespace.maximum[source]#
deepmd._vendors.ndtensorflow._namespace.minimum[source]#
deepmd._vendors.ndtensorflow._namespace.multiply[source]#
deepmd._vendors.ndtensorflow._namespace.not_equal[source]#
deepmd._vendors.ndtensorflow._namespace.positive[source]#
deepmd._vendors.ndtensorflow._namespace.sin[source]#
deepmd._vendors.ndtensorflow._namespace.sinh[source]#
deepmd._vendors.ndtensorflow._namespace.square[source]#
deepmd._vendors.ndtensorflow._namespace.sqrt[source]#
deepmd._vendors.ndtensorflow._namespace.subtract[source]#
deepmd._vendors.ndtensorflow._namespace.tan[source]#
deepmd._vendors.ndtensorflow._namespace.expm1(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.tanh(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.remainder(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.bitwise_and(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.bitwise_left_shift(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.bitwise_invert(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.bitwise_or(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.bitwise_right_shift(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.bitwise_xor(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.ceil(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.floor(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.trunc(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.copysign(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.hypot(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.imag(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.isfinite(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.isinf(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.isnan(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.log(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.floor_divide(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.log1p(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.log2(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.log10(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.negative(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.pow(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.logaddexp(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.nextafter(x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.real(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.reciprocal(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.round(x: deepmd._vendors.ndtensorflow._array.Array, /, *, decimals: int = 0) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.sign(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.signbit(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.clip(x: deepmd._vendors.ndtensorflow._array.Array, /, min: deepmd._vendors.ndtensorflow._array.Array | complex | None = None, max: deepmd._vendors.ndtensorflow._array.Array | complex | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.all(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.any(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.max(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.min(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.mean(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, dtype: DType | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.prod(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, dtype: DType | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.sum(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, dtype: DType | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.var(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, correction: float = 0.0, dtype: DType | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.std(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, correction: float = 0.0, dtype: DType | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.cumulative_sum(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | None = None, dtype: DType | None = None, include_initial: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.cumulative_prod(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | None = None, dtype: DType | None = None, include_initial: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.diff(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int = -1, n: int = 1, prepend: deepmd._vendors.ndtensorflow._array.Array | None = None, append: deepmd._vendors.ndtensorflow._array.Array | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.argsort(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int = -1, descending: bool = False, stable: bool = True) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.sort(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int = -1, descending: bool = False, stable: bool = True) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.take(x: deepmd._vendors.ndtensorflow._array.Array, indices: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.take_along_axis(x: deepmd._vendors.ndtensorflow._array.Array, indices: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int = -1) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.matmul(x1: deepmd._vendors.ndtensorflow._array.Array, x2: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.matrix_transpose(x: deepmd._vendors.ndtensorflow._array.Array, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.tensordot(x1: deepmd._vendors.ndtensorflow._array.Array, x2: deepmd._vendors.ndtensorflow._array.Array, /, *, axes: int | tuple[collections.abc.Sequence[int], collections.abc.Sequence[int]] = 2) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.vecdot(x1: deepmd._vendors.ndtensorflow._array.Array, x2: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int = -1) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.broadcast_shapes(*shapes: tuple[int, ...]) → tuple[int, ...][source]#
deepmd._vendors.ndtensorflow._namespace.broadcast_to(x: deepmd._vendors.ndtensorflow._array.Array, /, shape: tuple[int, ...]) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.broadcast_arrays(*arrays: deepmd._vendors.ndtensorflow._array.Array) → tuple[deepmd._vendors.ndtensorflow._array.Array, ...][source]#
deepmd._vendors.ndtensorflow._namespace.concat(arrays: tuple[deepmd._vendors.ndtensorflow._array.Array, ...] | list[deepmd._vendors.ndtensorflow._array.Array], /, *, axis: int | None = 0) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.expand_dims(x: deepmd._vendors.ndtensorflow._array.Array, /, axis: int | tuple[int, ...]) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.flip(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.meshgrid(*arrays: deepmd._vendors.ndtensorflow._array.Array, indexing: Literal['xy', 'ij'] = 'xy') → tuple[deepmd._vendors.ndtensorflow._array.Array, ...][source]#
deepmd._vendors.ndtensorflow._namespace.moveaxis(x: deepmd._vendors.ndtensorflow._array.Array, /, source: int | collections.abc.Sequence[int], destination: int | collections.abc.Sequence[int]) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.permute_dims(x: deepmd._vendors.ndtensorflow._array.Array, /, axes: tuple[int, ...]) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.transpose(x: deepmd._vendors.ndtensorflow._array.Array, /, axes: tuple[int, ...] | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.einsum(subscripts: str, *operands: deepmd._vendors.ndtensorflow._array.Array) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.repeat(x: deepmd._vendors.ndtensorflow._array.Array, repeats: int | deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.reshape(x: deepmd._vendors.ndtensorflow._array.Array, /, shape: tuple[int, ...], *, copy: bool | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.roll(x: deepmd._vendors.ndtensorflow._array.Array, /, shift: int | tuple[int, ...], axis: int | tuple[int, ...] | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.squeeze(x: deepmd._vendors.ndtensorflow._array.Array, /, axis: int | tuple[int, ...]) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.stack(arrays: tuple[deepmd._vendors.ndtensorflow._array.Array, ...] | list[deepmd._vendors.ndtensorflow._array.Array], /, *, axis: int = 0) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.tile(x: deepmd._vendors.ndtensorflow._array.Array, repetitions: tuple[int, ...], /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.unstack(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int = 0) → tuple[deepmd._vendors.ndtensorflow._array.Array, ...][source]#
deepmd._vendors.ndtensorflow._namespace.argmax(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.argmin(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.count_nonzero(x: deepmd._vendors.ndtensorflow._array.Array, /, *, axis: int | tuple[int, ...] | None = None, keepdims: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.nonzero(x: deepmd._vendors.ndtensorflow._array.Array, /) → tuple[deepmd._vendors.ndtensorflow._array.Array, ...][source]#
deepmd._vendors.ndtensorflow._namespace.searchsorted(x1: deepmd._vendors.ndtensorflow._array.Array, x2: deepmd._vendors.ndtensorflow._array.Array, /, *, side: Literal['left', 'right'] = 'left', sorter: deepmd._vendors.ndtensorflow._array.Array | None = None) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.where(condition: deepmd._vendors.ndtensorflow._array.Array, x1: Any, x2: Any, /) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.unique_all(x: deepmd._vendors.ndtensorflow._array.Array) → UniqueAllResult[source]#
deepmd._vendors.ndtensorflow._namespace.unique_counts(x: deepmd._vendors.ndtensorflow._array.Array) → UniqueCountsResult[source]#
deepmd._vendors.ndtensorflow._namespace.unique_inverse(x: deepmd._vendors.ndtensorflow._array.Array) → UniqueInverseResult[source]#
deepmd._vendors.ndtensorflow._namespace.unique_values(x: deepmd._vendors.ndtensorflow._array.Array) → deepmd._vendors.ndtensorflow._array.Array[source]#
deepmd._vendors.ndtensorflow._namespace.isin(x1: deepmd._vendors.ndtensorflow._array.Array | int, x2: deepmd._vendors.ndtensorflow._array.Array | int, /, *, invert: bool = False) → deepmd._vendors.ndtensorflow._array.Array[source]#