python torch库

作者：Angel★狂龙发布日期:2025-09-26 浏览:54

import torch

# 创建一个随机张量
x = torch.rand(5, 3)
print("随机张量 x:")
print(x)

# 创建一个全为0的张量
y = torch.zeros(5, 3, dtype=torch.float32)
print("\n全为0的张量 y:")
print(y)

# 创建一个全为1的张量
z = torch.ones(5, 3, dtype=torch.float32)
print("\n全为1的张量 z:")
print(z)

# 张量加法
result = x + y
print("\nx + y 的结果:")
print(result)

# 张量乘法
result_mul = x * z
print("\nx * z 的结果:")
print(result_mul)

# 将张量转换为numpy数组
numpy_x = x.numpy()
print("\n张量 x 转换为 numpy 数组:")
print(numpy_x)

# 解释说明:
# 1. `torch.rand` 用于创建一个指定形状的随机张量。
# 2. `torch.zeros` 和 `torch.ones` 分别用于创建全为0和全为1的张量。
# 3. 张量之间的加法和乘法可以直接使用 `+` 和 `*` 操作符。
# 4. 使用 `.numpy()` 方法可以将 PyTorch 张量转换为 NumPy 数组，方便与其他库进行交互。

上一篇：python event

下一篇：python tcp服务器