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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
use imp_prelude::*;
use error::{ShapeError, ErrorKind, from_kind};
pub fn stack<'a, A, D>(axis: Axis, arrays: &[ArrayView<'a, A, D>])
-> Result<OwnedArray<A, D>, ShapeError>
where A: Copy,
D: RemoveAxis
{
if arrays.len() == 0 {
return Err(from_kind(ErrorKind::Unsupported));
}
let mut res_dim = arrays[0].dim();
if axis.axis() >= res_dim.ndim() {
return Err(from_kind(ErrorKind::OutOfBounds));
}
let common_dim = res_dim.remove_axis(axis);
if arrays.iter().any(|a| a.dim().remove_axis(axis) != common_dim) {
return Err(from_kind(ErrorKind::IncompatibleShape));
}
let stacked_dim = arrays.iter()
.fold(0, |acc, a| acc + a.shape().axis(axis));
res_dim.set_axis(axis, stacked_dim);
let size = res_dim.size();
let mut v = Vec::with_capacity(size);
unsafe {
v.set_len(size);
}
let mut res = try!(OwnedArray::from_vec_dim(res_dim, v));
{
let mut assign_view = res.view_mut();
for array in arrays {
let len = array.shape().axis(axis);
let (mut front, rest) = assign_view.split_at(axis, len);
front.assign(array);
assign_view = rest;
}
}
Ok(res)
}
#[macro_export]
macro_rules! stack {
($axis:expr, $( $array:expr ),+ ) => {
ndarray::stack($axis, &[ $(ndarray::ArrayView::from(&$array) ),* ]).unwrap()
}
}