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\]\])) [@浙大疏锦行](https://blog.csdn.net/weixin_45655710 "@浙大疏锦行") print("\\n加法结果:") print(result)