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