def updateOutput(self, input):
self._lazyInit()
dimension = self._getPositiveDimension(input)
torch.min(self._output, self._indices, input, dimension)
if input.dim() > 1:
self.output.set_(self._output.select(dimension, 0))
else:
self.output.set_(self._output)
return self.output
评论列表
文章目录