本文主要是介绍pytorch基础: torch.unbind()，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
1. torch.unbind 作用
-
说明：
移除指定维后，返回一个元组，包含了沿着指定维切片后的各个切片。 -
参数：
- tensor(Tensor) – 输入张量
- dim(int) – 删除的维度
2. 案例
案例1
x = torch.rand(1,80,3,360,360)y= x.unbind(dim=2)print("y0 shape",y[0].shape)print("y1 shape",y[1].shape)print("y2 shape",y[2].shape)
- 将
shape大小为(1,80,2,360,360)的x,沿着dim为2的维度切片。 - 此时会移除
dim为2的维度，得到由3个 元素大小为(1,80,360,360)的tensor组成的元组。 - 元组中tensor个数，和指定切片对应维度的值相等。
x = torch.rand(1,80,3,360,360)a =torch.cat(x.unbind(dim=2),1)a.shape
a.shape: torch.Size([1,240,360,360])
- 切片后得到包含3个Tensor的元组，每个tensor大小为
(1,80,360,360) - 3个tensor沿dim为1进行concate, 因此得到tensor大小为
(1,240,360,360)
这篇关于pytorch基础: torch.unbind()的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
