PyTorch torch.ravel 函数
Pytorch torch 参考手册torch.ravel 是 PyTorch 中用于将张量展平为一维的函数。返回的是输入张量的视图（如果可能），而不是副本。
函数定义
torch.ravel(input)
参数说明：
input: 输入张量
使用示例
实例
import torch
# 创建 2x3 矩阵
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 展平为一维
y = torch.ravel(x)
print(y)
# 创建 2x3 矩阵
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 展平为一维
y = torch.ravel(x)
print(y)
输出结果为：
tensor([1, 2, 3, 4, 5, 6])
实例
import torch
# 创建三维张量
x = torch.arange(12).reshape(2, 3, 2)
# 展平为一维
y = torch.ravel(x)
print(y)
# 创建三维张量
x = torch.arange(12).reshape(2, 3, 2)
# 展平为一维
y = torch.ravel(x)
print(y)
输出结果为：
tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])