Main Function
Shumai offers a wide range of tensor operations and computational capabilities, including tensor creation, reshaping, slicing, indexing, mathematical operations, linear algebra, convolution, pooling, and more. Additionally, Shumai supports automatic differentiation, making it easy to compute gradients and perform backpropagation.