xtructure.core.xtructure_numpy package

Subpackages

• xtructure.core.xtructure_numpy.dataclass_ops package
  • Subpackages
    • xtructure.core.xtructure_numpy.dataclass_ops.unique_ops package
      • Submodules
      • xtructure.core.xtructure_numpy.dataclass_ops.unique_ops.benchmark_unqiue_ops module
      • xtructure.core.xtructure_numpy.dataclass_ops.unique_ops.legacy_unique_ops module
      • xtructure.core.xtructure_numpy.dataclass_ops.unique_ops.optimized_unique_ops module
      • Module contents
  • Submodules
  • xtructure.core.xtructure_numpy.dataclass_ops.batch_ops module
    • block()
    • column_stack()
    • concat()
    • dstack()
    • hstack()
    • pad()
    • split()
    • stack()
    • take()
    • take_along_axis()
    • tile()
    • vstack()
  • xtructure.core.xtructure_numpy.dataclass_ops.comparison_ops module
    • allclose()
    • equal()
    • isclose()
    • not_equal()
  • xtructure.core.xtructure_numpy.dataclass_ops.fill_ops module
    • full_like()
    • ones_like()
    • zeros_like()
  • xtructure.core.xtructure_numpy.dataclass_ops.logical_ops module
    • update_on_condition()
    • where()
    • where_no_broadcast()
  • xtructure.core.xtructure_numpy.dataclass_ops.shape_ops module
    • atleast_1d()
    • atleast_2d()
    • atleast_3d()
    • broadcast_arrays()
    • broadcast_to()
    • expand_dims()
    • flatten()
    • moveaxis()
    • repeat()
    • reshape()
    • squeeze()
    • swapaxes()
    • transpose()
  • xtructure.core.xtructure_numpy.dataclass_ops.spatial_ops module
    • flip()
    • roll()
    • rot90()
  • xtructure.core.xtructure_numpy.dataclass_ops.type_ops module
    • astype()
    • can_cast()
    • result_type()
  • Module contents
    • allclose()
    • astype()
    • atleast_1d()
    • atleast_2d()
    • atleast_3d()
    • block()
    • broadcast_arrays()
    • broadcast_to()
    • can_cast()
    • column_stack()
    • concat()
    • dstack()
    • equal()
    • expand_dims()
    • flatten()
    • flip()
    • full_like()
    • hstack()
    • isclose()
    • moveaxis()
    • not_equal()
    • ones_like()
    • pad()
    • repeat()
    • reshape()
    • result_type()
    • roll()
    • rot90()
    • split()
    • squeeze()
    • stack()
    • swapaxes()
    • take()
    • take_along_axis()
    • tile()
    • transpose()
    • unique_mask()
    • update_on_condition()
    • vstack()
    • where()
    • where_no_broadcast()
    • zeros_like()

Submodules

xtructure.core.xtructure_numpy.array_ops module

Module contents

xtructure.core.xtructure_numpy.allclose(a: Any, b: Any, rtol: float = 1e-05, atol: float = 1e-08, equal_nan: bool = False) → bool

xtructure.core.xtructure_numpy.astype(x, dtype, /, *, copy: bool = False, device=None)

xtructure.core.xtructure_numpy.atleast_1d(*arys: Any) → Any

xtructure.core.xtructure_numpy.atleast_2d(*arys: Any) → Any

xtructure.core.xtructure_numpy.atleast_3d(*arys: Any) → Any

xtructure.core.xtructure_numpy.block(arrays: Any) → Any

xtructure.core.xtructure_numpy.broadcast_arrays(*args: Any) → list[Any]

xtructure.core.xtructure_numpy.broadcast_to(array, shape, *, out_sharding=None)

xtructure.core.xtructure_numpy.can_cast(from_: Any, to: Any, casting: str = 'safe') → bool

xtructure.core.xtructure_numpy.column_stack(tup: Sequence[Any]) → Any

xtructure.core.xtructure_numpy.concat(arrays, /, *, axis: int | None = 0)

xtructure.core.xtructure_numpy.concatenate(arrays, axis: int | None = 0, dtype: Any | None = None)

xtructure.core.xtructure_numpy.dstack(tup: Sequence[Any], dtype: Any = None) → Any

xtructure.core.xtructure_numpy.equal(x, y, /)

xtructure.core.xtructure_numpy.expand_dims(a, axis: int | Sequence[int])

xtructure.core.xtructure_numpy.flatten(array: Any, order: str = 'C') → Any

xtructure.core.xtructure_numpy.flip(m: Any, axis: int | Sequence[int] | None = None) → Any

xtructure.core.xtructure_numpy.full_like(a, fill_value, dtype: Any | None = None, shape: Any = None, *, device=None)

xtructure.core.xtructure_numpy.hstack(tup: Sequence[Any], dtype: Any = None) → Any

xtructure.core.xtructure_numpy.isclose(a: Any, b: Any, rtol: float = 1e-05, atol: float = 1e-08, equal_nan: bool = False) → Any

xtructure.core.xtructure_numpy.moveaxis(a: Any, source: int | Sequence[int], destination: int | Sequence[int]) → Any

xtructure.core.xtructure_numpy.not_equal(x, y, /)

xtructure.core.xtructure_numpy.ones_like(a, dtype=None, shape=None, *, device=None, out_sharding=None)

xtructure.core.xtructure_numpy.pad(array, pad_width, mode: str | Any = 'constant', **kwargs)

xtructure.core.xtructure_numpy.ravel(a, order: str = 'C', *, out_sharding=None)

xtructure.core.xtructure_numpy.repeat(a, repeats, axis: int | None = None, *, total_repeat_length: int | None = None, out_sharding=None)

xtructure.core.xtructure_numpy.reshape(a, shape, order: str = 'C', *, copy: bool | None = None, out_sharding=None)

xtructure.core.xtructure_numpy.result_type(*args: Any) → Any

xtructure.core.xtructure_numpy.roll(a: Any, shift: int | Sequence[int], axis: int | Sequence[int] | None = None) → Any

xtructure.core.xtructure_numpy.rot90(m: Any, k: int = 1, axes: tuple[int, int] = (0, 1)) → Any

xtructure.core.xtructure_numpy.split(ary, indices_or_sections, axis: int = 0)

xtructure.core.xtructure_numpy.squeeze(a, axis: int | Sequence[int] | None = None)

xtructure.core.xtructure_numpy.stack(arrays, axis: int = 0, out: None = None, dtype: Any | None = None)

xtructure.core.xtructure_numpy.swapaxes(a, axis1: int, axis2: int)

xtructure.core.xtructure_numpy.take(a, indices, axis: int | None = None, out=None, mode: str | None = None, unique_indices: bool = False, indices_are_sorted: bool = False, fill_value=None)

xtructure.core.xtructure_numpy.take_along_axis(arr, indices, axis: int | None = -1, mode=None, fill_value=None)

xtructure.core.xtructure_numpy.tile(A, reps)

xtructure.core.xtructure_numpy.transpose(a, axes: Sequence[int] | None = None)

xtructure.core.xtructure_numpy.unique_mask(val: Any, key: Any | None = None, filled: Any | None = None, key_fn: Any | None = None, batch_len: int | None = None, return_index: bool = False, return_inverse: bool = False) → Any

xtructure.core.xtructure_numpy.update_on_condition(dataclass_instance, indices, condition, values_to_set)

xtructure.core.xtructure_numpy.vstack(tup: Sequence[Any], dtype: Any = None) → Any

xtructure.core.xtructure_numpy.where(condition, x=None, y=None, /, *, size=None, fill_value=None)

xtructure.core.xtructure_numpy.where_no_broadcast(condition: Any, x: Any, y: Any) → Any

xtructure.core.xtructure_numpy.zeros_like(a, dtype=None, shape=None, *, device=None, out_sharding=None)