pytorch view(-1),pytorch x.view

  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的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。

留言与评论(共有 条评论)
   
验证码: