mpsgraph 0.2.0

Modern Rust bindings for Apple's Metal Performance Shaders Graph framework
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
66
67
68
69
70
71
72
73
74
75
76

use crate::{Graph, Tensor};
use objc2::{extern_methods, msg_send, rc::Retained};
use objc2_foundation::NSString;

impl Graph {
    /// Creates a one-dimensional slice along a given axis.
    ///
    /// # Arguments
    ///
    /// * `tensor` – Source tensor to slice.
    /// * `dimension_index` – Axis along which to slice.
    /// * `start` – Starting index (negative values count from the end of the
    ///   dimension).
    /// * `length` – Number of elements to keep.
    /// * `name` – Optional debug label.
    ///
    /// # Returns
    ///
    /// A [`Tensor`] representing the slice.
    pub fn slice(
        &self,
        tensor: &Tensor,
        dimension_index: u64,
        start: i64,
        length: i64,
        name: Option<&str>,
    ) -> Retained<Tensor> {
        unsafe {
            msg_send![
                self,
                sliceTensor: tensor,
                dimension: dimension_index,
                start: start,
                length: length,
                name: name.map(NSString::from_str).as_deref(),
            ]
        }
    }

    /// Creates an N-dimensional slice using *start* and *size* tensors.
    ///
    /// The operation starts at `start_tensor` and keeps `size_tensor` elements
    /// along each dimension, following the semantics of TensorFlow’s *Slice*
    /// op.
    ///
    /// # Arguments
    ///
    /// * `tensor` – Tensor to slice.
    /// * `start_tensor` – Per-dimension starting indices.
    /// * `size_tensor` – Per-dimension slice lengths.
    /// * `squeeze_mask` – Bitmask of dimensions to remove from the result.
    /// * `name` – Optional debug label.
    ///
    /// # Returns
    ///
    /// A [`Tensor`] containing the slice.
    pub fn slice_with_tensors(
        &self,
        tensor: &Tensor,
        start_tensor: &Tensor,
        size_tensor: &Tensor,
        squeeze_mask: u32,
        name: Option<&str>,
    ) -> Retained<Tensor> {
        unsafe {
            msg_send![
                self,
                sliceTensor: tensor,
                startTensor: start_tensor,
                sizeTensor: size_tensor,
                squeezeMask: squeeze_mask,
                name: name.map(NSString::from_str).as_deref(),
            ]
        }
    }
}