hpt 0.1.3

High Performance Tensor (HPT) - A fast, efficient, and user-friendly tensor computation library for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::Tensor;
use hpt_allocator::{
    traits::{Allocator, AllocatorOutputRetrive},
    Cpu,
};
use hpt_common::error::base::TensorError;
use hpt_traits::{ops::cumulative::CumulativeOps, tensor::CommonBounds};

impl<T: CommonBounds, const DEVICE: usize, Al> CumulativeOps for Tensor<T, Cpu, DEVICE, Al>
where
    Al: Allocator,
    Al::Output: AllocatorOutputRetrive,
{
    fn cumsum<A: Into<Option<i64>>>(&self, axis: A) -> Result<Self, TensorError> {
        Ok(self.inner.cumsum(axis)?.into())
    }
    fn cumprod<A: Into<Option<i64>>>(&self, axis: A) -> Result<Self, TensorError> {
        Ok(self.inner.cumprod(axis)?.into())
    }
}