Module prelude_dev 

Re-exports

pub use crate::DeviceCpu;

pub use crate::tensor::reduction::TensorSumBoolAPI;

pub use crate::storage::adv_indexing::*;

pub use crate::storage::assignment::*;

pub use crate::storage::combined_trait::*;

pub use crate::storage::conversion::*;

pub use crate::storage::creation::*;

pub use crate::storage::data::*;

pub use crate::storage::device::*;

pub use crate::storage::matmul::*;

pub use crate::storage::operators::*;

pub use crate::storage::reduction::*;

pub use crate::device_cpu_serial::device::*;

pub use crate::tensor::asarray::*;

pub use crate::tensor::creation::*;

pub use crate::tensor::creation_from_tensor::*;

pub use crate::tensor::device_conversion::*;

pub use crate::tensor::ext_conversion::*;

pub use crate::tensor::iterator_axes::*;

pub use crate::tensor::iterator_elem::*;

pub use crate::tensor::manuplication::*;

pub use crate::tensor::ownership_conversion::*;

pub use crate::tensor::tensor_mutable::*;

pub use crate::prelude::rstsr_traits::*;

pub use crate::feature_rayon::device::*;

pub use crate::device_faer::device::*;

pub use crate::tensorbase::*;

Modules

TensorCopyPolicy

The policy of copying tensor.

broadcast

Layout broadcasting.

dim

indexer

iterator

Layout (double-ended) iterator.

layoutbase

Layout of tensor.

matmul

Layout manuplication for matmul and other linalg operations

rearrangement

Layout rearrangement.

reshape

Auxiliary function for reshaping a tensor.

shape

slice

stride

vec

A contiguous growable array type with heap-allocated contents, written Vec<T>.

Macros

duplicate

Duplicates the given code and substitutes specific identifiers for different code snippets in each duplicate.

format

Creates a String using interpolation of runtime expressions.

impl_from_tuple_to_axes_index

izip

Create an iterator running multiple iterators in lockstep.

rstsr_assert

rstsr_assert_eq

rstsr_errcode

rstsr_error

rstsr_invalid

rstsr_pattern

rstsr_raise

s

slice

Generate slice with into support and optional parameters.

vec

Creates a Vec containing the arguments.

Structs

Box

A pointer type that uniquely owns a heap allocation of type T.

IndexedIterLayout

IterLayoutColMajor

Layout iterator (column-major).

IterLayoutRowMajor

Layout iterator (row-major).

Layout

Layout of tensor.

LayoutMatMulConfig

ParIterRSTSR

PhantomData

Zero-sized type used to mark things that “act like” they own a T.

Slice

Slicing for python (numpy) convention; somehow similar to Rust’s range.

Stride

String

A UTF-8–encoded, growable string.

ThreadPool

Represents a user-created thread pool.

Vec

A contiguous growable array type, written as Vec<T>, short for ‘vector’.

Enums

AxesIndex

Enum for Axes indexing

BroadcastType

Error

FlagDiag

FlagOrder

The order of the tensor.

FlagSide

FlagSymm

FlagTrans

FlagUpLo

Indexer

IterLayout

MatMulType

Rules of matmul.

TensorIterOrder

The policy of the tensor iterator.

Traits

ChangeableDefault

Debug

? formatting.

DimAPI

DimBaseAPI

DimBroadcastableAPI

DimDevAPI

DimIntoAPI

DimLayoutContigAPI

DimShapeAPI

DimStrideAPI

Display

Format trait for an empty format, {}.

IndexerDynamicAPI

IndexerLargerOneAPI

IndexerPreserveAPI

IndexerSmallerOneAPI

IterSplitAtAPI

Itertools

An Iterator blanket implementation that provides extra adaptors and methods.

LayoutMatMulAPI

ToString

A trait for converting a value to a String.

Write

A trait for writing or formatting into Unicode-accepting buffers or streams.

Functions

aligned_alloc

Create an uninitialized vector with the given size and alignment.

aligned_uninitialized_vec^⚠

Create an conditionally aligned uninitialized vector with the given size.

allclose_f64

Compare two tensors with f64 data type.

assign_arbitary_cpu_rayon

assign_arbitary_cpu_serial

assign_cpu_rayon

assign_cpu_serial

broadcast_layout

Layout broadcasting.

broadcast_layout_to_first

Layout broadcasting.

broadcast_shape

Shape broadcasting.

fill_cpu_rayon

fill_cpu_serial

fingerprint

Get a somehow unique fingerprint of a tensor.

gemm_ix2_naive_cpu_rayon

gemm_naive_cpu_serial

gemv_naive_cpu_serial

gevm_naive_cpu_serial

greedy_layout

This function will return a f-prefer layout that make minimal memory accessing efforts (pointers will not frequently back-and-forth).

index_select_cpu_rayon

index_select_cpu_serial

inner_dot_naive_cpu_rayon

inner_dot_naive_cpu_serial

inner_pack_tril_c_contig

inner_pack_tril_general

inner_pack_triu_c_contig

inner_pack_triu_general

inner_unpack_tril_c_contig

inner_unpack_tril_general

inner_unpack_triu_c_contig

inner_unpack_triu_general

layout_col_major_dim_dispatch_1

layout_col_major_dim_dispatch_2

layout_col_major_dim_dispatch_3

layout_col_major_dim_dispatch_2diff

layout_col_major_dim_dispatch_par_1

layout_col_major_dim_dispatch_par_2

layout_col_major_dim_dispatch_par_3

layout_col_major_dim_dispatch_par_2diff

layout_for_array_copy

Return a layout that is suitable for array copy.

layout_reshapeable

Check if a tensor can be reshaped to a new shape without explicitly copying underlying data.

matmul_naive_cpu_serial

op_muta_func_cpu_rayon

op_muta_func_cpu_serial

op_muta_numb_func_cpu_rayon

op_muta_numb_func_cpu_serial

op_muta_refb_func_cpu_rayon

op_muta_refb_func_cpu_serial

op_mutc_numa_refb_func_cpu_rayon

op_mutc_numa_refb_func_cpu_serial

op_mutc_refa_numb_func_cpu_rayon

op_mutc_refa_numb_func_cpu_serial

op_mutc_refa_refb_func_cpu_rayon

op_mutc_refa_refb_func_cpu_serial

orderchange_out_c2r_ix2_cpu_rayon

Change order (row/col-major) a matrix out-place using a naive algorithm.

orderchange_out_c2r_ix2_cpu_serial

Change order (row/col-major) a matrix out-place using a naive algorithm.

orderchange_out_r2c_ix2_cpu_rayon

Change order (row/col-major) a matrix out-place using a naive algorithm.

orderchange_out_r2c_ix2_cpu_rayon_no_pool

Change order (row/col-major) a matrix out-place using a naive algorithm.

orderchange_out_r2c_ix2_cpu_serial

Change order (row/col-major) a matrix out-place using a naive algorithm.

pack_tri_cpu_rayon

pack_tri_cpu_serial

reduce_all_arg_cpu_rayon

reduce_all_arg_cpu_serial

reduce_all_cpu_rayon

reduce_all_cpu_serial

reduce_all_unraveled_arg_cpu_rayon

reduce_all_unraveled_arg_cpu_serial

reduce_axes_arg_cpu_rayon

reduce_axes_arg_cpu_serial

reduce_axes_cpu_rayon

reduce_axes_cpu_serial

reduce_axes_difftype_cpu_rayon

reduce_axes_difftype_cpu_serial

reduce_axes_unraveled_arg_cpu_rayon

reduce_axes_unraveled_arg_cpu_serial

reshape_substitute_negatives

Check -1 in shape and substitute it with the correct value.

reversed_permute

Reversed permutation indices.

translate_to_col_major

Translate multiple layouts to column-major iteration.

translate_to_col_major_unary

Translate one layout to column-major iteration.

translate_to_col_major_with_contig

This function will return minimal dimension layout, that the first axis is f-contiguous.

tril_cpu_serial

tril_ix2_cpu_serial

triu_cpu_serial

triu_ix2_cpu_serial

unaligned_uninitialized_vec^⚠

Create an unaligned uninitialized vector with the given size.

uninitialized_vec^⚠

Create an uninitialized vector with the given size.

unpack_tri_cpu_rayon

unpack_tri_cpu_serial

unrolled_reduce

Fold over the manually unrolled xs with f

update_layout_by_shape

Type Aliases

Ix

Fixed size dimension.

Ix0

Ix1

Ix2

Ix3

Ix4

Ix5

Ix6

Ix7

Ix8

Ix9

IxD

IxDyn

Dynamic size dimension.

Result

SliceI

In most cases, we will use isize for indexing.

TensorDiag

TensorOrder

TensorSide

TensorSymm

TensorTrans

TensorUpLo

Attribute Macros

duplicate_item

Duplicates the item and substitutes specific identifiers for different code snippets in each duplicate.

substitute_item

Substitutes specific identifiers for different code snippets.

Derive Macros

Debug

Derive macro generating an impl of the trait Debug.