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

use hpt_common::error::base::TensorError;
use hpt_traits::ops::binary::TensorDot;
use hpt_types::type_promote::NormalOut;

use crate::{tensor_base::_Tensor, Tensor};
use hpt_traits::tensor::CommonBounds;
impl<A, B> TensorDot<Tensor<B>> for Tensor<A>
where
    _Tensor<A>: TensorDot<_Tensor<B>, Output = _Tensor<<A as NormalOut<B>>::Output>>,
    A: CommonBounds + NormalOut<B>,
    B: CommonBounds,
    <A as NormalOut<B>>::Output: CommonBounds,
{
    type Output = Tensor<<A as NormalOut<B>>::Output>;

    fn tensordot<const N: usize>(
        &self,
        rhs: &Tensor<B>,
        axes: ([i64; N], [i64; N]),
    ) -> std::result::Result<Self::Output, TensorError> {
        let res = self.inner.tensordot(rhs.inner.as_ref(), axes)?;
        Ok(res.into())
    }
}