如何减少PyTorch张量打印时元素间的多余空格？

打印PyTorch张量时元素间存在大量空格，导致内容难以清晰读取，当前使用的打印设置如下：

torch.set_printoptions(sci_mode=False, precision=4, linewidth=200, profile="full")
np.set_printoptions(suppress=True, precision=4, linewidth=200)

PyTorch张量显示效果：

同形状NumPy数组因无多余空格更易读，显示效果：

解决方法 #

PyTorch默认会对张量元素做右对齐排版，这是导致多余空格的主要原因，通过调整torch.set_printoptions的参数即可实现紧凑输出：

• 添加align=None参数，关闭强制列对齐，避免为了对齐而添加多余空格
• 可选：调整separator参数自定义元素分隔符（默认是, ，可改为,进一步压缩）
• 建议移除profile="full"，该配置会启用更宽松的打印格式，换成默认的profile="default"或直接省略

修改后的示例代码：

torch.set_printoptions(
    sci_mode=False,
    precision=4,
    linewidth=200,
    align=None,
    separator=', ',  # 可根据需求改为','
    profile="default"
)
np.set_printoptions(suppress=True, precision=4, linewidth=200)

调整后，PyTorch张量的输出风格会和NumPy数组一致，元素间无多余空格，更易读取。

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