MrJun's Blog

Deep Learning Framework API Record

Tensorflow1.x

tf.nn.top_k(
    input,
    k=1,
    sorted=True,
    name=None
)
# 找到输入的张量的最后的一个维度的最大的k个值和它的下标。
# 参考链接:https://blog.csdn.net/m0_37393514/article/details/82380109

tf.tile(
    input,     #输入
    multiples,  #同一维度上复制的次数
    name=None
)
# 平铺,用于在同一维度上的复制。
# 参考链接:https://blog.csdn.net/weixin_41089007/article/details/93924098

tf.gather_nd(
    params, 
    indices, 
    name=None
)
# 根据indices描述的索引,提取params上的元素, 重新构建一个tensor
# 参考链接:https://blog.csdn.net/G66565906/article/details/84949512

tf.constant(
    value,
    dtype=None,
    shape=None,
    name='Const',
    verify_shape=False
)
# 创建常量
# 参考链接:https://blog.csdn.net/csdn_jiayu/article/details/82155224