python打印变量懒人版 locals

python print的时候如果想要print x=xvalue格式变量名得输出两边，下面是懒人搞法，但后缀有些长，如果有更好的方法欢迎推荐：

import torch
import numpy as np


def get_attn_pad_mask(seq_q, seq_k):
    '''
    seq_q: [batch_size, len_q]
    seq_k: [batch_size, len_k]
    seq_len could be src_len or it could be tgt_len
    seq_len in seq_q and seq_len in seq_k maybe not equal
    '''
    batch_size, len_q = seq_q.size()
    batch_size, len_k = seq_k.size()
    # eq(zero) is PAD token
    pad_attn_mask = seq_k.data.eq(0).unsqueeze(1)  # [batch_size, 1, len_k], False is masked
    return pad_attn_mask.expand(batch_size, len_q, len_k)  # [batch_size, len_q, len_k]

def get_attn_subsequence_mask(seq):
    '''
    seq: [batch_size, tgt_len]
    '''
    attn_shape = [seq.size(0), seq.size(1), seq.size(1)]
    subsequence_mask = np.triu(np.ones(attn_shape), k=1) # Upper triangular matrix
    subsequence_mask = torch.from_numpy(subsequence_mask).byte()
    return subsequence_mask

#不太work，输出的变量名成val了
def print_val(val):
    tmp = locals()
    print('{val}'.format(**{k: '{}={}'.format(k, v) for (k, v) in locals().items()}))

if __name__ == '__main__':
    seq = torch.Tensor([[2.0,3.0,5.0],[1.0,2.0,0.0]])
    pad_mask = get_attn_pad_mask(seq,seq)
    print(f'pad_mask={pad_mask}')
    print('pad_mask={}'.format(pad_mask))
    print('{pad_mask}'.format(**{k: '{}={}'.format(k, v) for (k, v) in locals().items()}))