PyTorch API 函数文档

PyTorch 提供了丰富的 API 函数，这些函数可以让你更方便地使用 PyTorch 进行深度学习。以下是一些常用的 API 函数及其简要说明。

常用 API 函数列表

• torch.tensor(data, dtype=None, device=None, requires_grad=False): 创建一个 PyTorch 张量。
• torch.nn.Module: 定义一个神经网络模型。
• torch.optim: 定义一个优化器。
• torch.nn.functional: 提供了各种神经网络函数。

例子

以下是一个简单的例子，展示如何使用 torch.tensor 创建一个张量：

import torch

# 创建一个形状为 (3, 2) 的浮点型张量
x = torch.tensor([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]])
print(x)

扩展阅读

更多关于 PyTorch API 函数的详细说明，请参考PyTorch 官方文档.

图片展示

PyTorch Logo