rstsr-core 0.7.0

An n-Dimension Rust Tensor Toolkit
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use crate::prelude_dev::*;

#[duplicate_item(
    OpAPI           ;
   [OpAcosAPI      ];
   [OpAcoshAPI     ];
   [OpAsinAPI      ];
   [OpAsinhAPI     ];
   [OpAtanAPI      ];
   [OpAtanhAPI     ];
   [OpCeilAPI      ];
   [OpConjAPI      ];
   [OpCosAPI       ];
   [OpCoshAPI      ];
   [OpExpAPI       ];
   [OpExpm1API     ];
   [OpFloorAPI     ];
   [OpInvAPI       ];
   [OpIsFiniteAPI  ];
   [OpIsInfAPI     ];
   [OpIsNanAPI     ];
   [OpLogAPI       ];
   [OpLog1pAPI     ];
   [OpLog2API      ];
   [OpLog10API     ];
   [OpReciprocalAPI];
   [OpRoundAPI     ];
   [OpSignBitAPI   ];
   [OpSinAPI       ];
   [OpSinhAPI      ];
   [OpSquareAPI    ];
   [OpSqrtAPI      ];
   [OpTanAPI       ];
   [OpTanhAPI      ];
   [OpTruncAPI     ];
   [OpAbsAPI       ];
   [OpImagAPI      ];
   [OpRealAPI      ];
   [OpSignAPI      ];
)]
pub trait OpAPI<T, D>
where
    D: DimAPI,
    Self: DeviceAPI<T> + DeviceAPI<MaybeUninit<Self::TOut>>,
{
    type TOut;

    fn op_muta_refb(
        &self,
        a: &mut <Self as DeviceRawAPI<MaybeUninit<Self::TOut>>>::Raw,
        la: &Layout<D>,
        b: &<Self as DeviceRawAPI<T>>::Raw,
        lb: &Layout<D>,
    ) -> Result<()>;

    fn op_muta(&self, a: &mut <Self as DeviceRawAPI<MaybeUninit<Self::TOut>>>::Raw, la: &Layout<D>) -> Result<()>;
}

// Python Array API specifications (2023.1)

// not implemented types
// DeviceBitwiseInvertAPI, (implemented in Not)
// DeviceLogicalNotAPI, (implemented in Not)
// DeviceNegativeAPI, (implemented in Neg)
// DevicePositiveAPI, (not implemented)