metrics.py 文件源码

python
阅读 31 收藏 0 点赞 0 评论 0

项目:tefla 作者: openAGI 项目源码 文件源码
def accumulate_strings(values, name="strings"):
    """Accumulates strings into a vector.

    Args:
      values: A 1-d string tensor that contains values to add to the accumulator.

    Returns:
      A tuple (value_tensor, update_op).
    """
    tf.assert_type(values, tf.string)
    strings = tf.Variable(
        name=name,
        initial_value=[],
        dtype=tf.string,
        trainable=False,
        collections=[],
        validate_shape=True)
    value_tensor = tf.identity(strings)
    update_op = tf.assign(
        ref=strings, value=tf.concat([strings, values], 0), validate_shape=False)
    return value_tensor, update_op
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号