二维tensor转为一维元素个数不变。用pytorch写

可以使用view()函数将二维tensor转换为一维，注意要保证元素个数不变。

示例代码：

import torch

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

# 使用view()函数将二维tensor转换为一维
y = x.view(-1)  # -1表示自动计算维度

print("x:\n", x)
print("y:\n", y)

输出结果：

x:
 tensor([[1, 2, 3],
         [4, 5, 6]])
y:
 tensor([1, 2, 3, 4, 5, 6])