如何将函数应用于二维张量的每个元素？

import numpy as np

# 定义一个示例函数，将元素加倍
def double(x):
    return 2 * x

# 创建一个二维张量
tensor = np.array([[1, 2, 3], [4, 5, 6]])

# 获取张量的形状
shape = tensor.shape

# 创建一个与原始张量形状相同的全零张量，用于存储结果
result = np.zeros(shape)

# 遍历每个元素并应用函数
for i in range(shape[0]):
    for j in range(shape[1]):
        result[i, j] = double(tensor[i, j])

# 打印结果
print(result)