pytorch view(-1),pytorch x.view
本文主要介绍。pytorch中的view()函数,主要介绍手动调整大小和自动调整大小两种方法。以下具体方法可以供你参考。
00 -1010一、一般用法(手动调节大小)二、使用方法。特殊用途:参数-1(自动调整大小)
目录
View()相当于重塑、调整和重新调整张量的形状。
进口火炬
a1=torch.arange(0,16)
打印(a1)
#张量([ 0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15])
a2=a1.view(8,2)
a3=a1.view(2,8)
a4=a1.view(4,4)
打印(a2)
#张量([[ 0,1],
# [ 2, 3],
# [ 4, 5],
# [ 6, 7],
# [ 8, 9],
# [10, 11],
# [12, 13],
# [14, 15]])
打印(a3)
#张量([[ 0,1,2,3,4,5,6,7],
# [ 8, 9, 10, 11, 12, 13, 14, 15]])
打印(a4)
#张量([[ 0,1,2,3],
# [ 4, 5, 6, 7],
# [ 8, 9, 10, 11],
# [12, 13, 14, 15]])
一、普通用法 (手动调整size)
view中的一个参数设置为-1,这意味着自动调整该维度中的元素数量,以确保元素总数保持不变。
v1=torch.arange(0,16)
打印(v1)
#张量([ 0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15])
v2=v1.view(-1,16)
v2
#张量([[ 0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]])
v2=v1.view(-1,8)
v2
#张量([[ 0,1,2,3,4,5,6,7],
# [ 8, 9, 10, 11, 12, 13, 14, 15]])
v2=v1.view(-1,4)
v2
#张量([[ 0,1,2,3],
# [ 4, 5, 6, 7],
# [ 8, 9, 10, 11],
# [12, 13, 14, 15]])
v2=v1.view(-1,2)
v2
#张量([[ 0,1],
# [ 2, 3],
# [ 4, 5],
# [ 6, 7],
# [ 8, 9],
# [10, 11],
# [12, 13],
# [14, 15]])
v3=v1.view(4*4,-1)
v3
#张量([[ 0],
# [ 1],
# [ 2],
# [ 3],
# [ 4],
# [ 5],
# [ 6],
# [ 7],
# [ 8],
# [ 9],
# [10],
# [11],
# [12],
# [13],
# [14],
# [15]])
关于的用法,这篇文章就到这里了。pytorch中的view()函数。有关pytorch的更多信息。view()函数,请搜索热门IT软件开发工作室之前的文章或继续浏览下面的相关文章。希望大家以后多多支持热门IT软件开发工作室!
郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。