利用np.triu()函数生成一个上三角矩阵
import numpy as np
size=512
shape=(1,size,size)
one_ = np.ones(shape)
print(one_)
mask = np.triu(np.ones(shape), k=1).astype("uint8") # 生产上三角矩阵
print(mask)
import numpy as np
size=512
shape=(1,size,size)
one_ = np.ones(shape)
print(one_)
mask = np.triu(np.ones(shape), k=1).astype("uint8") # 生产上三角矩阵
print(mask)