Python学习Day48

学习来源：浙大疏锦行

1 import torch
 
# 生成标量（0 维张量）
scalar = torch.randn(())
print(f"标量: {scalar}, 形状: {scalar.shape}")
 
# 生成向量（1 维张量）
vector = torch.randn(5)  # 长度为 5 的向量
print(f"向量: {vector}, 形状: {vector.shape}")
 
# 生成矩阵（2 维张量）
matrix = torch.randn(3, 4)  # 3 行 4 列的矩阵
print(f"矩阵：{matrix}, 矩阵形状: {matrix.shape}")
2 import torch
 
# 加法广播示例
a = torch.tensor([[1, 2, 3], [4, 5, 6]])  # 形状: (2, 3)
b = torch.tensor([10, 20, 30])             # 形状: (3,)
result_add = a + b  
print(f"加法广播结果: {result_add}")
 
# 乘法广播示例
c = torch.tensor([[1, 2], [3, 4], [5, 6]])  # 形状: (3, 2)
d = torch.tensor([10, 20])                  # 形状: (2,)
result_mul = c * d  
print(f"乘法广播结果: {result_mul}")