在 Caffe 中实现“转置”层时出错

Question

我正在尝试使用 Python 层在 Caffe 中实现转置功能。 下面是相同的代码。

但是，在Reshape() 方法中抛出boost::python::error_already_set 的实例后，它会抛出错误“终止调用。

有人可以说明我做错了什么吗？

import caffe
import numpy as np

class transpose(caffe.Layer):

    def setup(self, bottom, top):
        assert len(bottom) == 1,            'requires a single layer.bottom'
        assert bottom[0].data.ndim == 2,    'requires matrix data'
        assert len(top) == 1,               'requires a single layer.top'

    def reshape(self, bottom, top):
        top[0].reshape((bottom[0].data.shape[1], bottom[0].data.shape[0]))

    def forward(self, bottom, top):
        top[0].data = np.transpose(bottom[0].data)

    def backward(self, top, propagate_down, bottom):
        pass

谢谢你， 维杰塔。

Answer 1

我认为你 reshapeing 不正确。
试试：

def reshape(self, bottom, top):
  top[0].reshape(bottom[0].data.shape[1], bottom[0].data.shape[0])

Reshape 的shape 参数不作为元组给出，而是作为单独的参数给出。