欢迎访问宙启技术站
智能推送

使用tensor_vstack()函数在Python中进行张量的垂直堆叠操作

发布时间:2023-12-17 16:40:47

在Python中使用tensor_vstack()函数可以对张量进行垂直堆叠操作。垂直堆叠是指将多个张量按垂直方向堆叠在一起,产生一个尺寸更大的新张量。tensor_vstack()函数将会把输入的张量按照垂直方向进行堆叠,并返回一个新的张量。

tensor_vstack()函数的语法如下:

torch.tensor_vstack(tensors, dtype=None, device=None, requires_grad=False) → Tensor

参数说明:

- tensors: 一个表示输入张量的元组或列表。

- dtype: 生成的张量的数据类型,默认为None,表示与输入张量的类型一致。

- device: 生成的张量存放的设备,默认为None,表示与输入张量存放的设备一致。

- requires_grad: 是否为生成的张量开启梯度跟踪,默认为False。

下面是一个使用tensor_vstack()函数进行张量垂直堆叠的例子:

import torch

# 创建两个张量
tensor1 = torch.tensor([[1, 2, 3],
                        [4, 5, 6]])
tensor2 = torch.tensor([[7, 8, 9],
                        [10, 11, 12]])

# 使用tensor_vstack()对张量进行垂直堆叠
result = torch.tensor_vstack((tensor1, tensor2))

print(result)

输出结果为:

tensor([[ 1,  2,  3],
        [ 4,  5,  6],
        [ 7,  8,  9],
        [10, 11, 12]])

在上述例子中,首先我们创建了两个张量tensor1tensor2。然后使用tensor_vstack()函数对这两个张量进行了垂直堆叠操作,并将结果保存到变量result中。最后我们打印输出了结果。

可以看到,tensor_vstack()函数将tensor1tensor2按垂直方向堆叠在一起,生成了一个新的尺寸更大的张量。