如何创建缩放PyTorch张量的比例矩阵，然后如何使用它？

import torch

# 创建一个3x3的比例矩阵，比例因子为2
scale_factor = 2
scale_matrix = torch.diag(torch.tensor([scale_factor] * 3))

# 创建一个3x3的PyTorch张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 将比例矩阵应用于PyTorch张量
scaled_tensor = torch.matmul(scale_matrix, tensor)

print("缩放比例矩阵：")
print(scale_matrix)
print("原始张量：")
print(tensor)
print("缩放后的张量：")
print(scaled_tensor)

缩放比例矩阵：
tensor([[2, 0, 0],
        [0, 2, 0],
        [0, 0, 2]])
原始张量：
tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])
缩放后的张量：
tensor([[ 2,  4,  6],
        [ 8, 10, 12],
        [14, 16, 18]])