torsh-tensor 0.1.2

Tensor implementation for ToRSh with PyTorch-compatible API
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
//! Tensor manipulation operations for combining, splitting, and transforming tensors
//!
//! This module provides PyTorch-compatible tensor manipulation operations including:
//! - Concatenation: cat, stack
//! - Splitting: split, chunk
//! - Flipping: flip, fliplr, flipud, rot90
//! - Repeating: tile, repeat, repeat_interleave
//! - Rolling: roll
//! - Dimension manipulation: movedim, moveaxis, swapaxes, swapdims
//! - Shape transformation: unflatten
//! - Advanced indexing: take_along_dim

mod core_ops;
mod dim_ops;

#[cfg(test)]
mod tests;