Skip to main content

torsh_python/tensor/
mod.rs

1//! Tensor module - PyTorch-compatible tensor operations in Python bindings
2//!
3//! This module provides a modular structure for tensor operations:
4//! - `core` - Core PyTensor struct and basic operations
5//! - `creation` - Tensor creation functions (zeros, ones, randn, etc.)
6
7pub mod core;
8pub mod creation;
9
10// Re-export the main types
11pub use core::PyTensor;
12pub use creation::register_creation_functions;