怎么使用 PyTorch 的 shape 或者 size 函数查看 tensor 的大小

PyTorch 的 `shape` 和 `size` 函数可以用来查看 tensor 的大小。这两个函数的作用是相同的，都可以返回 tensor 的尺寸大小。下面是使用这两个函数的示例代码：

import torch

# 创建一个 2x3 的 tensor
x = torch.Tensor([[1, 2, 3], [4, 5, 6]])

# 使用 shape 函数查看 tensor 的大小
print(x.shape)  # 输出 torch.Size([2, 3])

# 使用 size 函数查看 tensor 的大小
print(x.size())  # 输出 torch.Size([2, 3])

在上述代码中，我们首先使用 `torch.Tensor` 创建了一个 2 行 3 列的 tensor，然后分别使用 `shape` 和 `size` 函数查看了 tensor 的大小。两个函数的返回结果都是 `torch.Size([2, 3])`，表示这个 tensor 有 2 行 3 列。需要注意的是，`shape` 和 `size` 函数返回的都是一个 `torch.Size` 类型的对象，它可以像元组一样使用，例如，可以使用 `x.shape[0]` 来获取 tensor 的行数。