Pytorch之view及view_as使用详解

作者:啧啧啧biubiu 时间:2023-10-09 23:10:34 

view()函数是在torch.Tensor.view()下的一个函数,可以有tensor调用,也可以有variable调用。

其作用在于返回和原tensor数据个数相同,但size不同的tensor

【Numpy中的size是元素个数,但是在Pytorch中size等价为Numpy中的shape】

view函数的-1参数的作用在于基于另一参数,自动计算该维度的大小

很重要的一点

view函数只能由于contiguous的张量上,具体而言,就是在内存中连续存储的张量。

具体而言,可以参看 https://www.jb51.net/article/177564.htm

所以,当tensor之前调用了transpose, permute函数就会是tensor内存中变得不再连续,就不能调用view函数。

所以,应该提前做tensor.contiguous()的操作

view函数与Pytorch0.4中新增的reshape的区别

reshape函数调用是不依赖于tensor在内存中是不是连续的。

reshape ≈ tensor.contiguous().view

代码


import numpy as np
import torch
from torch.autograd import Variable

x = torch.Tensor(2,2,2)
print(x)

y = x.view(1,8)
print(y)

z = x.view(-1,4) # the size -1 is inferred from other dimensions
print(z)

t = x.view(8)
print(t)

输出


tensor([[[1.3712e-14, 6.4069e+02],
  [4.3066e+21, 1.1824e+22]],

[[4.3066e+21, 6.3828e+28],
  [3.8016e-39, 0.0000e+00]]])

#x.view(1,8)生成的是[1,8]的张量
tensor([[1.3712e-14, 6.4069e+02, 4.3066e+21, 1.1824e+22, 4.3066e+21, 6.3828e+28,
  3.8016e-39, 0.0000e+00]])

#x.view(-1,4)其中-1是在4下的另一个维度的大小,也就是8/4=2,所以生成的是[2,4]的张量
tensor([[1.3712e-14, 6.4069e+02, 4.3066e+21, 1.1824e+22],
 [4.3066e+21, 6.3828e+28, 3.8016e-39, 0.0000e+00]])

x.view(8)生成的是[8,]的张量,是个数组
tensor([1.3712e-14, 6.4069e+02, 4.3066e+21, 1.1824e+22, 4.3066e+21, 6.3828e+28,
 3.8016e-39, 0.0000e+00])

view_as

返回被视作与给定的tensor相同大小的原tensor。 等效于:


self.view(tensor.size())

具体用法为:

代码


a = torch.Tensor(2, 4)
b = a.view_as(torch.Tensor(4, 2))
print (b)

输出


tensor([[1.3712e-14, 6.4069e+02],
 [4.3066e+21, 1.1824e+22],
 [4.3066e+21, 6.3828e+28],
 [3.8016e-39, 0.0000e+00]])

来源:https://blog.csdn.net/qq_37385726/article/details/81738518

标签:Pytorch,view,view,as
0
投稿

猜你喜欢

  • vue 自定义icon图标的步骤

    2024-04-30 10:37:15
  • JavaScript实现优先级队列

    2024-05-02 16:19:06
  • mysql5在rhel5下乱码问题及解决方法

    2010-12-03 16:26:00
  • 一次Mysql死锁排查过程的全纪录

    2024-01-16 20:35:53
  • Python基础Lists和tuple实例详解

    2021-09-27 04:11:36
  • 解决Keras中Embedding层masking与Concatenate层不可调和的问题

    2022-06-02 17:02:16
  • 利用Python生成Excel炫酷图表

    2023-03-09 21:52:00
  • oracle 性能优化建议小结

    2010-04-22 16:32:00
  • golang协程关闭踩坑实战记录

    2024-02-13 00:02:57
  • Python中if __name__ == '__main__'作用解析

    2021-04-14 09:02:42
  • 使用Keras 实现查看model weights .h5 文件的内容

    2023-10-18 03:07:16
  • debug case: onsubmit=return false依然能提交

    2009-01-18 13:07:00
  • mysql 设置默认的时间值

    2024-01-18 02:49:46
  • 解决import tensorflow as tf 出错的原因

    2023-06-26 23:53:21
  • 在FLASH中调用ASP的方法

    2009-03-09 18:32:00
  • python __init__与 __new__的区别

    2022-05-04 01:27:57
  • python字典快速保存于读取的方法

    2022-02-18 02:22:11
  • pytest conftest.py文件的使用讲解

    2023-10-16 14:22:44
  • 通过gradio和摄像头获取照片和视频实现过程

    2023-07-08 18:02:30
  • django用户登录和注销的实现方法

    2021-01-17 13:55:40
  • asp之家 网络编程 m.aspxhome.com