[ torch 参数更多 ]torch.set_printoptions¶
torch.set_printoptions¶
torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None, sci_mode=None)
paddle.set_printoptions¶
paddle.set_printoptions(precision=None, threshold=None, edgeitems=None, sci_mode=None, linewidth=None)
PyTorch 相比 Paddle 支持更多其他参数，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| precision | precision | 浮点数的小数位数, PyTorch 默认值为 4，Paddle 默认为 8。 |
| threshold | threshold | 打印的元素个数上限，默认值为 1000。 |
| edgeitems | edgeitems | 以缩略形式打印时左右两边的元素个数，默认值为 3。 |
| linewidth | linewidth | 每行的字符数，默认值为 80。 |
| sci_mode | sci_mode | 是否以科学计数法打印，PyTorch 默认根据网络自动选择， Paddle 默认值为 False。 |
| profile | - | 预设风格，支持 default, short, full。 Paddle 无此参数， 需要转写。 |
转写示例¶
profile:预设风格，设置为 default。¶
# PyTorch 写法
torch.set_printoptions(profile='default')
# Paddle 写法
paddle.set_printoptions(precision=4, threshold=1000, edgeitems=3, linewidth=80)
profile:预设风格，设置为 short。¶
# PyTorch 写法
torch.set_printoptions(profile='short')
# Paddle 写法
paddle.set_printoptions(precision=2, threshold=1000, edgeitems=2, linewidth=80)
profile:预设风格，设置为 full。¶
# PyTorch 写法
torch.set_printoptions(profile='full')
# Paddle 写法
paddle.set_printoptions(precision=4, threshold=1000000, edgeitems=3, linewidth=80)