正确的方法是将100x9广播到100x9x1x1 numpy数组，以便在Caffe中进行计算

Question

我正在尝试使用python包装器将自己的数据输入到caffe模型中。我将HDF5中的数据读取为尺寸为100x9的numpy数组。但是对于模型的输入，我使用以下代码：

input_ = np.zeros((100,9,1,1), dtype=np.float32)
net.forward(**{net.inputs[0]:input_})

所以基本上我需要从100x9数组中填写input_。

Answer 1

下面是如何将100x9数组转换为100x9x1x1数组：

x = np.zeros((100,9))
y = x[:,:,np.newaxis,np.newaxis]