python unsqueeze

作者：放电的牛发布日期:2026-07-18 浏览:61

import torch

# 创建一个一维张量
x = torch.tensor([1, 2, 3, 4])

# 使用 unsqueeze 方法在第 0 维增加一个维度
x_unsqueezed = x.unsqueeze(0)

print("原始张量:", x)
print("增加维度后的张量:", x_unsqueezed)

# 解释说明:
# unsqueeze 方法用于在指定位置插入一个维度大小为 1 的维度。
# 在上面的例子中，我们在第 0 维插入了一个新的维度，使得张量从 (4,) 变为了 (1, 4)。
# 这在处理批量数据或需要匹配其他张量的维度时非常有用。

上一篇：python cffi

下一篇：python stringio