如何在TensorFlow中为张量右侧添加指定数量的零值

直接使用TensorFlow原生的tf.pad接口即可实现该需求，不需要自定义额外的复杂逻辑：

基础实现（对应你的示例场景） #

import tensorflow as tf

tn = tf.constant([1, 2])
# 1维张量右侧补3个零，paddings参数对应每个维度[左侧补零数量, 右侧补零数量]
tn = tf.pad(tn, paddings=[[0, 3]], mode='CONSTANT', constant_values=0)
print(tn)
# 输出：<tf.Tensor: shape=(5,), dtype=int32, numpy=array([1, 2, 0, 0, 0])>

封装为通用函数 #

如果你需要和示例中tf.add_zeros用法一致的工具函数，可以直接封装如下：

def add_zeros(tensor, num_zeros, position='right'):
    tensor_rank = tf.rank(tensor)
    # 初始化补零配置，默认仅在最后一个维度补零
    paddings = [[0, 0] for _ in range(tensor_rank)]
    if position == 'right':
        paddings[-1][1] = num_zeros
    elif position == 'left':
        paddings[-1][0] = num_zeros
    else:
        raise ValueError("position参数仅支持'right'或'left'")
    return tf.pad(tensor, paddings, mode='CONSTANT', constant_values=0)

# 测试调用
tn = tf.constant([1, 2])
tn = add_zeros(tn, 3, 'right')
print(tn)
# 输出：tf.Tensor([1 2 0 0 0], shape=(5,), dtype=int32)

高维张量适配 #

如果是高维张量，只需要调整paddings参数对应维度的补零配置即可，比如2维张量要在每一行的末尾补3个零，paddings配置为[[0, 0], [0, 3]]即可。

内容的提问来源于stack exchange，提问作者OK 400