zoukankan      html  css  js  c++  java
  • torch.squeeze函数解释,torch.FloatTensor()函数作用解释

    1. torch.squeeze(x,N)  主要对数据维度进行压缩

    torch.squeeze(x,N)
    #也可以写为格式 
    x.squeeze(dim=N)

    含义当N未给定时,去掉x中所有维度为1的维度,当N给定为某一数值时,去掉在这一数值指定位置的维度为1的维度。

    代码示例:

    2.torch.unsqueeze(x,dim=N)    数据维度进行扩充。通过dim指定位置,给指定位置加上维数为1的维度。

    代码示例:(x.size()同上)

    3.torch.FloatTensor([1,2])   ,先看官方文档:

    上图是pytorch1.7.0给出的官方文档,意思是torch.Tensor定义了十种tensor类型,分别如下....,其中就包含有32位浮点数的torch.FloatTensor。

    而torch.FloatTensor()的作用就是把给定的list或者numpy转换成浮点数类型的tensor。

    代码示例:

  • 相关阅读:
    深入源码之JDK Logging
    随机数生成器
    简单日志LogHelper
    Xml、Json序列化
    Redis封装之Set
    Redis封装之Hash
    Redis封装之String
    Redis之创建
    AngularJS初接触
    简单缓存Cache
  • 原文地址:https://www.cnblogs.com/yz-lucky77/p/13922110.html
Copyright © 2011-2022 走看看