def cumsum(x, axis=0):
"""Cumulative sum of the values in a tensor, alongside the specified axis.
# Arguments
x: A tensor or variable.
axis: An integer, the axis to compute the sum.
# Returns
A tensor of the cumulative sum of values of `x` along `axis`.
"""
axis = _normalize_axis(axis, ndim(x))
return tf.cumsum(x, axis=axis)
tensorflow_backend.py 文件源码
python
阅读 31
收藏 0
点赞 0
评论 0
评论列表
文章目录