Day.48

张量生成:

vector = torch.randn(5)

print(f"向量: {vector}, 形状: {vector.shape}")

张量,向量相加:

import torch

a = torch.tensor(\[10, 20, 30]) # 形状: (3, 1)

b = torch.tensor(1, 2, 3) # 形状: (3,)

result = a + b

print("原始张量a:")

print(a)

print("\n原始张量b:")

print(b)

print("\n广播后a的值扩展:")

print(torch.tensor(\[10, 10, 10,

20, 20, 20,

30, 30, 30]))

print("\n广播后b的值扩展:")

print(torch.tensor(\[1, 2, 3,

1, 2, 3,

1, 2, 3])) @浙大疏锦行

print("\n加法结果:")

print(result)