Python 将100x9广播到100x9x1x1 numpy阵列以在Caffe中进行计算的正确方法

Python 将100x9广播到100x9x1x1 numpy阵列以在Caffe中进行计算的正确方法,python,numpy,caffe,Python,Numpy,Caffe,我正在尝试使用python包装器将自己的数据输入到caffe模型中。我从HDF5读取的数据是一个尺寸为100x9的numpy数组。但对于模型的输入,我使用以下代码: input_ = np.zeros((100,9,1,1), dtype=np.float32) net.forward(**{net.inputs[0]:input_}) 因此,基本上我需要从100x9数组中填写输入。以下是如何将100x9数组转换为100x9x1x1数组: x = np.zeros((100,9)) y = x

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

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

因此,基本上我需要从100x9数组中填写输入。

以下是如何将100x9数组转换为100x9x1x1数组:

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