rstsr_core/
prelude.rs

1pub mod rstsr_traits {
2    pub use rstsr_common::prelude::rstsr_traits::*;
3
4    pub use crate::storage::conversion::DeviceChangeAPI;
5    pub use crate::storage::data::{DataAPI, DataCloneAPI, DataForceMutAPI, DataIntoCowAPI, DataMutAPI, DataOwnedAPI};
6    pub use crate::storage::device::{DeviceAPI, DeviceBaseAPI, DeviceRawAPI, DeviceStorageAPI};
7    pub use crate::tensor::asarray::AsArrayAPI;
8    pub use crate::tensor::creation::{
9        ArangeAPI, EmptyAPI, EmptyLikeAPI, EyeAPI, FromNestedArrayAPI, FullAPI, FullLikeAPI, LinspaceAPI, OnesAPI,
10        OnesLikeAPI, TrilAPI, TriuAPI, ZerosAPI, ZerosLikeAPI,
11    };
12    pub use crate::tensor::creation_from_tensor::{
13        ConcatAPI, DiagAPI, HStackAPI, MeshgridAPI, StackAPI, UnstackAPI, VStackAPI,
14    };
15    pub use crate::tensor::device_conversion::{TensorChangeFromDevice, TensorDeviceChangeAPI};
16    pub use crate::tensor::operators::op_binary_common::{
17        TensorATan2API, TensorCopySignAPI, TensorEqualAPI, TensorFloorDivideAPI, TensorGreaterAPI,
18        TensorGreaterEqualAPI, TensorHypotAPI, TensorLessAPI, TensorLessEqualAPI, TensorLogAddExpAPI, TensorMaximumAPI,
19        TensorMinimumAPI, TensorNotEqualAPI, TensorPowAPI,
20    };
21    pub use crate::tensor::operators::op_unary_common::{
22        TensorAbsAPI, TensorAcosAPI, TensorAcoshAPI, TensorAsinAPI, TensorAsinhAPI, TensorAtanAPI, TensorAtanhAPI,
23        TensorCeilAPI, TensorConjAPI, TensorCosAPI, TensorCoshAPI, TensorExpAPI, TensorExpm1API, TensorFloorAPI,
24        TensorImagAPI, TensorInvAPI, TensorIsFiniteAPI, TensorIsInfAPI, TensorIsNanAPI, TensorLog10API, TensorLog2API,
25        TensorLogAPI, TensorRealAPI, TensorRoundAPI, TensorSignAPI, TensorSignBitAPI, TensorSinAPI, TensorSinhAPI,
26        TensorSqrtAPI, TensorSquareAPI, TensorTanAPI, TensorTanhAPI, TensorTruncAPI,
27    };
28    pub use crate::tensor::ownership_conversion::{TensorIntoOwnedAPI, TensorViewAPI, TensorViewMutAPI};
29    pub use crate::tensor::reduction::TensorSumBoolAPI;
30
31    #[cfg(feature = "rayon")]
32    pub use crate::feature_rayon::DeviceRayonAPI;
33}
34
35pub mod rstsr_structs {
36    pub use rstsr_common::prelude::rstsr_structs::*;
37
38    pub use crate::device_cpu_serial::device::DeviceCpuSerial;
39    #[cfg(feature = "faer")]
40    pub use crate::device_faer::device::DeviceFaer;
41    pub use crate::DeviceCpu;
42
43    pub use crate::tensor::tensor_mutable::TensorMutable;
44    pub use crate::{
45        Tensor, TensorAny, TensorArc, TensorBase, TensorCow, TensorMut, TensorRef, TensorReference, TensorView,
46        TensorViewMut,
47    };
48
49    pub use crate::tensor::manuplication::exports::ReshapeArgs;
50}
51
52pub mod rstsr_funcs {
53    pub use crate::tensor::adv_indexing::{bool_select, bool_select_f, index_select, index_select_f, take, take_f};
54    pub use crate::tensor::asarray::{asarray, asarray_f};
55    pub use crate::tensor::creation::{
56        arange, arange_f, assume_init, assume_init_f, empty, empty_f, empty_like, empty_like_f, eye, eye_f, full,
57        full_f, full_like, full_like_f, linspace, linspace_f, ones, ones_f, ones_like, ones_like_f, tril, tril_f, triu,
58        triu_f, uninit, uninit_f, zeros, zeros_f, zeros_like, zeros_like_f,
59    };
60    pub use crate::tensor::creation_from_tensor::{
61        concat, concat_f, concatenate, concatenate_f, diag, diag_f, hstack, hstack_f, meshgrid, meshgrid_f, stack,
62        stack_f, unstack, unstack_f, vstack, vstack_f,
63    };
64    pub use crate::tensor::indexing::{
65        diagonal, diagonal_f, diagonal_mut, diagonal_mut_f, into_diagonal, into_diagonal_f, into_diagonal_mut,
66        into_diagonal_mut_f, into_slice, into_slice_f, slice, slice_f, slice_mut, slice_mut_f,
67    };
68    pub use crate::tensor::manuplication::exports::{
69        broadcast_arrays, broadcast_arrays_f, broadcast_into, broadcast_into_f, broadcast_shapes, broadcast_shapes_f,
70        broadcast_to, broadcast_to_f, change_contig, change_contig_f, change_layout, change_layout_f, change_prefer,
71        change_prefer_f, change_shape, change_shape_f, change_shape_with_args, change_shape_with_args_f, expand_dims,
72        expand_dims_f, flip, flip_f, into_broadcast, into_broadcast_f, into_compatible_shape, into_compatible_shape_f,
73        into_contig, into_contig_f, into_dim, into_dim_f, into_dyn, into_expand_dims, into_expand_dims_f, into_flip,
74        into_flip_f, into_layout, into_layout_f, into_moveaxis, into_moveaxis_f, into_permute_dims,
75        into_permute_dims_f, into_prefer, into_prefer_f, into_reverse_axes, into_shape, into_shape_f,
76        into_shape_with_args, into_shape_with_args_f, into_squeeze, into_squeeze_f, into_swapaxes, into_swapaxes_f,
77        into_transpose, into_transpose_f, into_unsqueeze, into_unsqueeze_f, moveaxis, moveaxis_f, permute_dims,
78        permute_dims_f, reshape, reshape_f, reshape_with_args, reshape_with_args_f, reshapeable_without_copy,
79        reverse_axes, squeeze, squeeze_f, swapaxes, swapaxes_f, to_broadcast, to_broadcast_f, to_compatible_shape,
80        to_compatible_shape_f, to_contig, to_contig_f, to_dim, to_dim_f, to_dyn, to_layout, to_layout_f, to_prefer,
81        to_prefer_f, to_shape, to_shape_f, to_shape_with_args, to_shape_with_args_f, transpose, transpose_f, unsqueeze,
82        unsqueeze_f,
83    };
84
85    // binary arithmetics
86    pub use crate::tensor::operators::exports::{
87        add, add_f, add_with_output, add_with_output_f, div, div_f, div_with_output, div_with_output_f, mul, mul_f,
88        mul_with_output, mul_with_output_f, rem, rem_f, rem_with_output, rem_with_output_f, sub, sub_f,
89        sub_with_output, sub_with_output_f,
90    };
91    // binary arithmetics with assignment
92    pub use crate::tensor::operators::exports::{
93        add_assign, add_assign_f, div_assign, div_assign_f, mul_assign, mul_assign_f, rem_assign, rem_assign_f,
94        sub_assign, sub_assign_f,
95    };
96    // binary bitwise
97    pub use crate::tensor::operators::exports::{
98        bitand, bitand_f, bitand_with_output, bitand_with_output_f, bitor, bitor_f, bitor_with_output,
99        bitor_with_output_f, bitxor, bitxor_f, bitxor_with_output, bitxor_with_output_f, shl, shl_f, shl_with_output,
100        shl_with_output_f, shr, shr_f, shr_with_output, shr_with_output_f,
101    };
102    // binary bitwise with assignment
103    pub use crate::tensor::operators::exports::{
104        bitand_assign, bitand_assign_f, bitor_assign, bitor_assign_f, bitxor_assign, bitxor_assign_f, shl_assign,
105        shl_assign_f, shr_assign, shr_assign_f,
106    };
107    // unary arithmetics
108    pub use crate::tensor::operators::exports::{neg, neg_f, not, not_f};
109    // unary common functions
110    pub use crate::tensor::operators::exports::{
111        abs, abs_f, acos, acos_f, acosh, acosh_f, asin, asin_f, asinh, asinh_f, atan, atan_f, atanh, atanh_f, ceil,
112        ceil_f, conj, conj_f, cos, cos_f, cosh, cosh_f, exp, exp_f, expm1, expm1_f, floor, floor_f, imag, imag_f, inv,
113        inv_f, is_finite, is_finite_f, is_inf, is_inf_f, is_nan, is_nan_f, log, log10, log10_f, log2, log2_f, log_f,
114        real, real_f, reciprocal, reciprocal_f, round, round_f, sign, sign_f, signbit, signbit_f, sin, sin_f, sinh,
115        sinh_f, sqrt, sqrt_f, square, square_f, tan, tan_f, tanh, tanh_f, trunc, trunc_f,
116    };
117    // binary common functions
118    pub use crate::tensor::operators::exports::{
119        atan2, atan2_f, copysign, copysign_f, eq, eq_f, equal, equal_f, equal_than, equal_than_f, floor_divide,
120        floor_divide_f, ge, ge_f, greater, greater_equal, greater_equal_f, greater_equal_to, greater_equal_to_f,
121        greater_f, greater_than, greater_than_f, gt, gt_f, hypot, hypot_f, le, le_f, less, less_equal, less_equal_f,
122        less_equal_to, less_equal_to_f, less_f, less_than, less_than_f, log_add_exp, log_add_exp_f, lt, lt_f, maximum,
123        maximum_f, minimum, minimum_f, ne, ne_f, nextafter, nextafter_f, not_equal, not_equal_f, not_equal_to,
124        not_equal_to_f, pow, pow_f,
125    };
126    // reduction
127    pub use crate::tensor::reduction::{
128        all, all_all, all_all_f, all_axes, all_axes_f, all_f, allclose, allclose_f, any, any_all, any_all_f, any_axes,
129        any_axes_f, any_f, argmax, argmax_all, argmax_all_f, argmax_axes, argmax_axes_f, argmax_f, argmin, argmin_all,
130        argmin_all_f, argmin_axes, argmin_axes_f, argmin_f, count_nonzero, count_nonzero_all, count_nonzero_all_f,
131        count_nonzero_axes, count_nonzero_axes_f, count_nonzero_f, l2_norm, l2_norm_all, l2_norm_all_f, l2_norm_axes,
132        l2_norm_axes_f, l2_norm_f, max, max_all, max_all_f, max_axes, max_axes_f, max_f, mean, mean_all, mean_all_f,
133        mean_axes, mean_axes_f, mean_f, min, min_all, min_all_f, min_axes, min_axes_f, min_f, prod, prod_all,
134        prod_all_f, prod_axes, prod_axes_f, prod_f, std, std_all, std_all_f, std_axes, std_axes_f, std_f, sum, sum_all,
135        sum_all_f, sum_axes, sum_axes_f, sum_f, unraveled_argmax, unraveled_argmax_all, unraveled_argmax_all_f,
136        unraveled_argmax_axes, unraveled_argmax_axes_f, unraveled_argmax_f, unraveled_argmin, unraveled_argmin_all,
137        unraveled_argmin_all_f, unraveled_argmin_axes, unraveled_argmin_axes_f, unraveled_argmin_f, var, var_all,
138        var_all_f, var_axes, var_axes_f, var_f,
139    };
140    // linalg (array-api's basic linalg operations, not the rstsr-linalg-traits)
141    pub use crate::tensor::linalg::exports::{
142        into_matrix_transpose, into_matrix_transpose_f, matmul, matmul_f, matmul_from, matmul_from_f,
143        matmul_with_output, matmul_with_output_f, matrix_transpose, matrix_transpose_f, vecdot, vecdot_f, vecdot_from,
144        vecdot_from_f,
145    };
146}
147
148pub mod rstsr_macros {
149    pub use crate::{allclose, tensor_from_nested};
150    pub use rstsr_common::prelude::rstsr_macros::*;
151}
152
153// final re-exports
154
155pub use rstsr_macros::*;
156pub use rstsr_structs::*;
157pub use rstsr_traits::*;
158
159#[allow(unused_imports)]
160pub mod rt {
161    pub use rstsr_common::prelude::rt::*;
162
163    pub use super::rstsr_funcs;
164    pub use super::rstsr_macros;
165    pub use super::rstsr_structs;
166    pub use super::rstsr_traits;
167
168    pub use super::rstsr_funcs::*;
169    pub use super::rstsr_macros::*;
170    pub use super::rstsr_structs::*;
171    pub use super::rstsr_traits::*;
172}
rstsr_core/prelude.rs

rstsr_core/
prelude.rs
