aoluming / cost_model Goto Github PK

x1=input.view(input.shape[0],input.shape[1],input.shape[2],input.shape[3]*input.shape[4])
x2=input.transpose(2,3)
x2 = x2.contiguous().view(x2.shape[0],x2.shape[1],x2.shape[2],x2.shape[3]*x2.shape[4])
·······
x3=input.transpose(2,4)
x3 = x3.contiguous().view(x3.shape[0], x3.shape[1], x3.shape[2], x3.shape[3] * x3.shape[4])
out3=self.conv(x3)
out3=self.batchnorm(out3)
out3=self.relu(out3)
out3 = out3.view(input.shape[0], input.shape[1], input.shape[2], int(input.shape[3] / self.stride),int(input.shape[4] / self.stride))