DL代码提升3

getattr() 函数用于返回一个对象属性值

setattr() 函数对应函数 getattr()，用于设置属性值

>>>class A(object):
     bar = 1
 
>>> a = A()
>>> getattr(a, 'bar')        # 获取属性 bar 值
1

>>> setattr(a, 'bar', 5)       # 设置属性 bar 值
>>> a.bar
5

可以使用zip函数来同时遍历两个数组

from collections import defaultdict

def count_same_elements(arr1, arr2):
    count_dict = defaultdict(int)
    for x, y in zip(arr1, arr2):
        if x == y:
            count_dict[x] += 1
    return dict(count_dict)
arr1 = [1, 2, 4, 4, 5]
arr2 = [1, 2, 4, 4, 6]
result_dict = count_same_elements(arr1, arr2)
print(result_dict)  # {1: 1, 2: 1, 4: 2}

nonzero()

会返回一个多维张量，其中每行表示一个值为1的元素的索引。我们将as_tuple参数设置为True，这样就可以将结果分解为单独的张量，而不是一个2D张量。这样可以取出非零元素对应位置的元素。