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
21
22
23
24
25
26
27
28
29
30
31

use crate::{backend::Cuda, tensor::Tensor, tensor_base::_Tensor};
use hpt_common::error::base::TensorError;
use hpt_traits::{ops::windows::WindowOps, tensor::CommonBounds};

use hpt_allocator::traits::{Allocator, AllocatorOutputRetrive};

impl<T, const DEVICE_ID: usize, Al> WindowOps for Tensor<T, Cuda, DEVICE_ID, Al>
where
    T: CommonBounds,
    _Tensor<T, Cuda, DEVICE_ID, Al>: WindowOps<Output = _Tensor<T, Cuda, DEVICE_ID, Al>>,
    _Tensor<T, Cuda, DEVICE_ID, Al>: Into<Tensor<T, Cuda, DEVICE_ID, Al>>,
    Al: Allocator,
    Al::Output: AllocatorOutputRetrive,
{
    type Output = Tensor<T, Cuda, DEVICE_ID, Al>;
    type Meta = T;
    #[track_caller]
    fn hamming_window(window_length: i64, periodic: bool) -> Result<Self::Output, TensorError> {
        Ok(_Tensor::<T, Cuda, DEVICE_ID, Al>::hamming_window(window_length, periodic)?.into())
    }

    #[track_caller]
    fn hann_window(window_length: i64, periodic: bool) -> Result<Self::Output, TensorError> {
        Ok(_Tensor::<T, Cuda, DEVICE_ID, Al>::hann_window(window_length, periodic)?.into())
    }

    #[track_caller]
    fn blackman_window(window_length: i64, periodic: bool) -> Result<Self::Output, TensorError> {
        Ok(_Tensor::<T, Cuda, DEVICE_ID, Al>::blackman_window(window_length, periodic)?.into())
    }
}