torch_backend.py 文件源码

def cumprod(x, axis=0):
    def _cumprod(x, axis=axis):
        y = torch.cumprod(x, axis)
        return y

    def _compute_output_shape(x, axis=axis):
        return _get_shape(x)

    return get_op(_cumprod, output_shape=_compute_output_shape, arguments=[axis])(x)

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~#