cidre 0.16.0

Apple frameworks bindings 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
32
33
34
35
36
37
38
39
40

use crate::{arc, mps::graph, ns, objc};

impl graph::Graph {
    #[objc::msg_send(meanOfTensor:axes:name:)]
    pub fn mean(
        &self,
        tensor: &graph::Tensor,
        axes: &ns::Array<ns::Number>,
        name: Option<&ns::String>,
    ) -> arc::R<graph::Tensor>;

    #[objc::msg_send(varianceOfTensor:axes:name:)]
    pub fn variance(
        &self,
        tensor: &graph::Tensor,
        axes: &ns::Array<ns::Number>,
        name: Option<&ns::String>,
    ) -> arc::R<graph::Tensor>;

    #[objc::msg_send(varianceOfTensor:meanTensor:axes:name:)]
    pub fn variance_mean(
        &self,
        tensor: &graph::Tensor,
        mean: &graph::Tensor,
        axes: &ns::Array<ns::Number>,
        name: Option<&ns::String>,
    ) -> arc::R<graph::Tensor>;

    #[objc::msg_send(normalizationWithTensor:meanTensor:varianceTensor:gammaTensor:betaTensor:epsilon:name:)]
    pub fn normalize(
        &self,
        tensor: &graph::Tensor,
        mean: &graph::Tensor,
        variance: &graph::Tensor,
        gamma: Option<&graph::Tensor>,
        beta: Option<&graph::Tensor>,
        epsilon: f32,
        name: Option<&ns::String>,
    ) -> arc::R<graph::Tensor>;
}