pytorch张量运算

创建张量，torch.tensor（）

import torch

# 创建一个标量（0维张量）
scalar = torch.tensor(5)

# 创建一个向量（1维张量）
vector = torch.tensor([1, 2, 3])

# 创建一个矩阵（2维张量）
matrix = torch.tensor([[1, 2], [3, 4]])

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

张量的加减乘除

###加法###
# 1.尺寸相同的张量相加，就是逐元素相加：
tensor1 = torch.tensor([[1, 2, 3], [4, 5, 6]])
tensor2 = torch.tensor([[7, 8, 9], [10, 11, 12]])
print(tensor1 + tensor2)
# 输出：
tensor([[ 8, 10, 12],
        [14, 16, 18]])

# 2.张量加数字就是每个元素上加这个数字：
tensor1 = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(tensor1