torsh-python 0.1.2

Python bindings for ToRSh - PyTorch-compatible deep learning in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

//! Tensor module - PyTorch-compatible tensor operations in Python bindings
//!
//! This module provides a modular structure for tensor operations:
//! - `core` - Core PyTensor struct and basic operations
//! - `creation` - Tensor creation functions (zeros, ones, randn, etc.)

pub mod core;
pub mod creation;

// Re-export the main types
pub use core::PyTensor;
pub use creation::register_creation_functions;