torch.tensor.expand - Python

torch.tensor.expand()是PyTorch中的一个方法，用于对张量进行扩张操作。该方法返回一个新的张量，其形状与原始张量形状不同。可以通过指定扩张的维度大小来控制新张量的形状。

语法

torch.Tensor.expand(size: Union[Tuple[int, ...], torch.Size]) -> torch.Tensor

参数

• size：一个元组或 torch.Size() 对象，指定输出张量的形状，它必须和输入张量的形状长度相同或更长。

返回值

返回一个新的扩展张量。

示例

import torch

# 创建一个 2x3 的张量
a = torch.Tensor([[1, 2, 3], [4, 5, 6]])
print(a.shape)  # 输出：torch.Size([2, 3])

# 对张量进行扩展操作
b = a.expand(2, 4)
print(b)
print(b.shape)  # 输出：torch.Size([2, 4, 3])

# 对张量进行扩展操作，也可以使用 size 参数
c = a.expand((2, 4))
print(c)
print(c.shape)  # 输出：torch.Size([2, 4, 3])

解释

上面的示例代码演示了如何使用 torch.tensor.expand() 对张量进行扩展操作。具体步骤如下：

1. 创建一个 2x3 的张量 a。
2. 对张量 a 进行扩展操作，将其扩展为一个 2x4 的张量 b。
3. 使用 size 参数对张量 a 进行扩展操作，将其扩展为一个 2x4 的张量 c。

总结

torch.tensor.expand() 方法可以用于对张量进行扩张操作，非常方便实用。在实际工作中，如果需要对张量进行扩张操作，可以使用该方法来实现。