Skip to content

BatchedArray

This page shows the supported operations for BatchedArray and BatchedArraySeq

Core functionalities

name BatchedArray BatchedArraySeq
batch_size ✅
data ✅
allclose ✅
allequal ✅
append ✅
chunk_along_batch ✅
clone ✅
extend ✅
get_num_minibatches ✅
index_select_along_batch ✅
permute_along_batch ✅
permute_along_batch_ ✅
select_along_batch ✅
shuffle_along_batch ✅
shuffle_along_batch_ ✅
slice_along_batch ✅
split_along_batch ✅
summary ✅
to_data ✅
to_minibatches ✅

Array creation

name redcat.ba BatchedArray BatchedArraySeq
array ✅ ❌
copy ✅
empty ✅
empty_like ✅ ✅
full ✅
full_like ✅ ✅
ones ✅
ones_like ✅ ✅
zeros ✅
zeros_like ✅ ✅

Logic functions

doc

name np redcat.ba BatchedArray BatchedArraySeq
Comparison
__eq__ ❌ ❌ ✅
__ge__ ❌ ❌ ✅
__gt__ ❌ ❌ ✅
__le__ ❌ ❌ ✅
__lt__ ❌ ❌ ✅
__ne__ ❌ ❌ ✅
allclose
array_equal
array_equiv
equal ✅
greater_equal ✅
greater ✅
less_equal ✅
less ✅
not_equal ✅
Array contents
isclose ✅
isfinite ✅
isinf ✅
isnan ✅
isnat ✅
isneginf ✅
isposinf ✅
Logical operations
logical_and ✅
logical_not ✅
logical_or ✅
logical_xor ✅

Array manipulation

doc

name redcat.np redcat.ba BatchedArray BatchedArraySeq
__getitem__ ✅
__setitem__ ✅
Joining arrays
concatenate ✅ ✅ ✅
concatenate_ ❌ ❌ ✅
concatenate_along_batch ❌ ✅ ✅
concatenate_along_batch_ ❌ ❌ ✅
concatenate_along_seq ❌ ❌
concatenate_along_seq_ ❌ ❌ ❌
Slicing arrays
chunk ❌ ✅
index_select ❌ ✅
select ❌ ✅
slice_along_axis ❌ ✅
slice_along_seq ❌ ❌
split_along_axis ❌ ✅
split_along_seq ❌ ❌
Rearranging elements
permute_along_axis ❌ ✅
permute_along_axis_ ❌ ✅
permute_along_seq ❌ ❌
permute_along_seq_ ❌ ❌
shuffle_along_axis ❌ ✅
shuffle_along_axis_ ❌ ✅
shuffle_along_seq ❌ ❌
shuffle_along_seq_ ❌ ❌

Math

doc

name np redcat.ba BatchedArray BatchedArraySeq
Arithmetic
__add__ ❌ ❌ ✅
__iadd__ ❌ ❌ ✅
__floordiv__ ❌ ❌ ✅
__ifloordiv__ ❌ ❌ ✅
__mul__ ❌ ❌ ✅
__imul__ ❌ ❌ ✅
__neg__ ❌ ❌ ✅
__sub__ ❌ ❌ ✅
__isub__ ❌ ❌ ✅
__truediv__ ❌ ❌ ✅
__itruediv__ ❌ ❌ ✅
add ✅ ✅ ✅
add_ ❌ ❌ ✅
divide ✅ ✅
divmod ✅ ✅
divmod_ ❌ ❌ ✅
floordiv ❌ ✅
floordiv_ ❌ ❌ ✅
floor_divide ✅ ✅
fmod ✅ ✅
fmod_ ❌ ❌ ✅
mul ❌ ✅
mul_ ❌ ❌ ✅
multiply ✅ ✅
sub ❌ ✅
sub_ ❌ ❌ ✅
substract ✅ ✅
truediv ❌ ❌ ✅
truediv_ ❌ ❌ ✅
true_divide ✅ ✅
Sums, products, differences
cumprod_along_batch ❌ ✅ ✅
cumprod_along_seq ❌ ❌
cumprod ✅ ✅ ✅
cumsum_along_batch ❌ ✅ ✅
cumsum_along_seq ❌ ❌
cumsum ✅ ✅ ✅
diff_along_batch ❌ ✅ ✅
diff_along_seq ❌ ❌
diff ✅ ✅ ✅
nancumprod_along_batch ❌ ✅ ✅
nancumprod_along_seq ❌ ❌
nancumprod ✅ ✅ ✅
nancumsum_along_batch ❌ ✅ ✅
nancumsum_along_seq ❌ ❌
nancumsum ✅ ✅ ✅
nanprod_along_batch ❌ ✅ ✅
nanprod_along_seq ❌ ❌
nanprod ✅ ✅ ✅
nansum_along_batch ❌ ✅ ✅
nansum_along_seq ❌ ❌
nansum ✅ ✅ ✅
prod_along_batch ❌ ✅ ✅
prod_along_seq ❌ ❌
prod ✅ ✅ ✅
sum_along_batch ❌ ✅ ✅
sum_along_seq ❌ ❌
sum ✅ ✅ ✅
trapz
Trigonometric functions
Hyperbolic functions
Exponents and logarithms
Rounding
Floating point routines
Rational routines
Extrema Finding
fmax ✅
fmin ✅
max_along_batch ❌ ✅ ✅
max_along_seq ❌ ❌
max ✅ ✅ ✅
maximum ✅
min_along_batch ❌ ✅ ✅
min_along_seq ❌ ❌
min ✅ ✅ ✅
minimum ✅
nanmax_along_batch ❌ ✅ ✅
nanmax_along_seq ❌ ❌
nanmax ✅ ✅ ✅
nanmin_along_batch ❌ ✅ ✅
nanmin_along_seq ❌ ❌
nanmin ✅ ✅ ✅

Sorting, searching, and counting

name np redcat.ba BatchedArray BatchedArraySeq
Sorting
argsort_along_batch ❌ ✅ ✅
argsort_along_seq ❌ ❌
argsort ✅ ✅ ✅
sort_along_batch ❌ ✅ ✅
sort_along_seq ❌ ❌
sort ✅ ✅ ✅
Searching
argmax_along_batch ❌ ✅ ✅
argmax_along_seq ❌ ❌
argmax ✅ ✅ ✅
argmin_along_batch ❌ ✅ ✅
argmin_along_seq ❌ ❌
argmin ✅ ✅ ✅
nanargmax_along_batch ❌ ✅ ✅
nanargmax_along_seq ❌ ❌
nanargmax ✅ ✅ ✅
nanargmin_along_batch ❌ ✅ ✅
nanargmin_along_seq ❌ ❌
nanargmin ✅ ✅ ✅

Statistics

name np redcat.ba BatchedArray BatchedArraySeq
mean_along_batch ❌ ✅ ✅
mean_along_seq ❌ ❌
mean ✅ ✅ ✅
median_along_batch ❌ ✅ ✅
median_along_seq ❌ ❌
median ✅ ✅ ✅
nanmean_along_batch ❌ ✅ ✅
nanmean_along_seq ❌ ❌
nanmean ✅ ✅ ✅
nanmedian_along_batch ❌ ✅ ✅
nanmedian_along_seq ❌ ❌
nanmedian ✅ ✅ ✅
nanquantile_along_batch ❌
nanquantile_along_seq ❌ ❌
nanquantile
nanstd_along_batch ❌
nanstd_along_seq ❌ ❌
nanstd
nanvar_along_batch ❌
nanvar_along_seq ❌ ❌
nanvar
quantile_along_batch ❌
quantile_along_seq ❌ ❌
quantile
std_along_batch ❌
std_along_seq ❌ ❌
std
var_along_batch ❌
var_along_seq ❌ ❌
var