pytorch基础 torch.unbind
发布时间
阅读量：
阅读量
1. torch.unbind 作用
说明 ：在删除特定维度之后，将生成一个元组，其中包含沿该维度分割得到的各个切片。
参数 ：
* tensor(Tensor) – 输入的张量
* dim(int) – 被移除的维度编号案例分析与应用验证
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变量，按照第2个维度进行切片操作。
- 此时将消除第2维度，从而生成一个包含3个元素的元组，每个元素的尺寸均为
(1,80,360,360)。 - 元组中
全部评论 (0)
还没有任何评论哟~