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