Xが4D、Yが2D、axisが0の場合に対応

sony · Apr 23, 2018 · 94e7037 · 94e7037
1 parent 98e01e5
commit 94e7037
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/python/test/function/test_broadcast_to.py b/python/test/function/test_broadcast_to.py
@@ -68,7 +68,8 @@ def ref_broadcast_to(x, y, axis):
                 return np.broadcast_to(t, x.shape)
         elif ys == 2:
             if axis == 0:
-                pass
+                t = y[:, :, np.newaxis, np.newaxis]
+                return np.broadcast_to(t, x.shape)
             elif axis == 1:
                 t = y[np.newaxis, :, :, np.newaxis]
                 return np.broadcast_to(t, x.shape)
@@ -87,6 +88,7 @@ def ref_broadcast_to(x, y, axis):
     ((2, 3, 4, 5), (3), 1),
     ((2, 3, 4, 5), (4), 2),
     ((2, 3, 4, 5), (5), 3),
+    ((2, 3, 4, 5), (2, 3), 0),
     ((2, 3, 4, 5), (3, 4), 1),
     ((2, 3, 4, 5), (4, 5), 2),
     #((2, 3, 4, 5), (2, 3, 4), 0),

diff --git a/src/nbla/function/broadcast_to.cpp b/src/nbla/function/broadcast_to.cpp
@@ -253,6 +253,10 @@ void BroadcastTo<T>::forward_impl(const Variables &inputs, const Variables &outp
 					switch(axis_) {
 						case 0:
 							// X: (2,3,4,5) Y: (2,3) axis=0
+							copy_buf_vertically_to_block(
+									z, y,
+									xs[0], ys[1],
+									xs[1], xs[2]*xs[3]);
 							break;
 						case 1:
 							{