JoinTable.py 文件源码

def updateOutput(self, input):
        dim = self._getPositiveDimension(input)

        for i in range(len(input)):
            currentOutput = input[i]
            if i == 0:
                size = list(currentOutput.size())
            else:
                size[dim] += currentOutput.size(dim)

        self.size = torch.Size(size)
        self.output.resize_(self.size)

        # TODO: use cat?
        offset = 0
        for i in range(len(input)):
            currentOutput = input[i]
            self.output.narrow(dim, offset, currentOutput.size(dim)).copy_(currentOutput)
            offset += currentOutput.size(dim)

        return self.output