cidre 0.11.8

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

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

impl graph::Graph {
    #[objc::msg_send(gatherWithUpdatesTensor:indicesTensor:axis:batchDimensions:name:)]
    pub fn gather(
        &self,
        updates: &graph::Tensor,
        indices: &graph::Tensor,
        axis: ns::UInteger,
        batch_dimensions: ns::UInteger,
        name: Option<&ns::String>,
    ) -> arc::R<graph::Tensor>;

    #[objc::msg_send(gatherAlongAxis:withUpdatesTensor:indicesTensor:name:)]
    pub fn gather_along_axis(
        &self,
        axis: ns::Integer,
        updates: &graph::Tensor,
        indices: &graph::Tensor,
        name: Option<&ns::String>,
    ) -> arc::R<graph::Tensor>;
}