张量生成:
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)