Fehler beim Implementieren von "transpose" Ebene in Caffe

Question

Ich versuche, die Transpose-Funktion in Caffe mit einer Python-Ebene zu implementieren. Unten ist der Code für das gleiche.

Es wird jedoch den Fehler zu werfen „nach dem Werfen einer Instanz boost::python::error_already_set bei Reshape() Methode namens beenden.

Kann jemand etwas Licht werfen auf das, was ich falsch mache?

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 

Danke, . Vijetha

Answer 1

ich glaube, Sie reshape falsch ing sind
Versuchen:.

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

shape Argumente für Reshape sind nicht als Tupel gegeben, sondern als separate Argumente.