以下是一个示例代码,用于生成一个不循环的二维阶梯函数张量:
import numpy as np
def create_staircase_tensor(n, m):
tensor = np.zeros((n, m))
for i in range(n):
for j in range(m):
if i <= j:
tensor[i, j] = 1
return tensor
n = 5
m = 5
tensor = create_staircase_tensor(n, m)
print(tensor)
这段代码使用了NumPy库来创建一个n×m的全零矩阵,然后根据阶梯函数的定义,对矩阵中的元素进行赋值。如果行索引小于等于列索引,则将元素的值设为1;否则,保持为0。最后,打印生成的阶梯函数张量。
运行代码输出结果如下:
[[1. 1. 1. 1. 1.]
[0. 1. 1. 1. 1.]
[0. 0. 1. 1. 1.]
[0. 0. 0. 1. 1.]
[0. 0. 0. 0. 1.]]
这个二维阶梯函数张量是不循环的,因为每一行的元素都是逐渐递增的。