pytorch中torch.flatten与torch.nn.flatten用法讲解

最新推荐文章于 2024-05-26 11:07:26 发布

郭庆汝

最新推荐文章于 2024-05-26 11:07:26 发布

阅读量775

点赞数

分类专栏：深度学习文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/guoqingru0311/article/details/127570470

版权

深度学习专栏收录该内容

51 篇文章 3 订阅

订阅专栏

pytorch中torch.flatten与torch.nn.flatten用法讲解

在这里插入图片描述

x = torch.ones(2, 2, 2, 2)

F = torch.nn.Flatten()
y = F(x)
print(y)
print(y.shape)
>>tensor([[1., 1., 1., 1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1., 1., 1., 1.]])
>>torch.Size([2, 8])

在这里插入图片描述

x = torch.ones(2, 2, 2, 2)

F = torch.nn.Flatten(2)
y = F(x)
print(y)
print(y.shape)
>>tensor([[[1., 1., 1., 1.],
         [1., 1., 1., 1.]],

        [[1., 1., 1., 1.],
         [1., 1., 1., 1.]]])
>>torch.Size([2, 2, 4])

在这里插入图片描述

x = torch.ones(2, 2, 2, 2)

F = torch.nn.Flatten(1, 2)
y = F(x)
print(y)
print(y.shape)
>>tensor([[[1., 1.],
         [1., 1.],
         [1., 1.],
         [1., 1.]],

        [[1., 1.],
         [1., 1.],
         [1., 1.],
         [1., 1.]]])
>>torch.Size([2, 4, 2])

在这里插入图片描述

t = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(t.shape)
>>torch.Size([2, 2, 2])

print(torch.flatten(t))
>>tensor([1, 2, 3, 4, 5, 6, 7, 8])

print(torch.flatten(t, 1))
>>tensor([[1, 2, 3, 4],
        [5, 6, 7, 8]])

print(torch.flatten(t, 0, 1).shape)
>>torch.Size([4, 2])

在这里插入图片描述

t = torch.tensor(1)
print("before flatten:")
print(t)
print(t.shape)
>>before flatten:
  tensor(1)
  torch.Size([])

print("\n")
print("after flatten:")
print(torch.flatten(t))
print(torch.flatten(t).shape)
>>after flatten:
  tensor([1])
  torch.Size([1])