Pytorch閱讀文件之flatten函式

2021-09-28 14:33:34 字數 3086 閱讀 8728

#展平乙個連續範圍的維度,輸出型別為tensor

torch.flatten(

input

, start_dim=

0, end_dim=-1

) → tensor

# parameters:input (tensor) – 輸入為tensor

#start_dim (int) – 展平的開始維度

#end_dim (int) – 展平的最後維度

#example

#乙個3x2x2的三維張量

>>

> t = torch.tensor([[

[1,2

],[3

,4]]

,[[5

,6],

[7,8

]],[

[9,10

],[11

,12]]

])#當開始維度為0,最後維度為-1,展開為一維

>>

> torch.flatten(t)

tensor([1

,2,3

,4,5

,6,7

,8,9

,10,11

,12])

#當開始維度為0,最後維度為-1,展開為3x4,也就是說第一維度不變,後面的壓縮

>>

> torch.flatten(t, start_dim=1)

tensor([[

1,2,

3,4]

,[5,

6,7,

8],[

9,10,

11,12]

])>>

> torch.flatten(t, start_dim=1)

.size(

)torch.size([3

,4])

#下面的和上面進行對比應該就能看出是,當鎖定最後的維度的時候

#前面的就會合併

>>

> torch.flatten(t, start_dim=

0, end_dim=1)

tensor([[

1,2]

,[3,

4],[

5,6]

,[7,

8],[

9,10]

,[11,

12]])

>>

> torch.flatten(t, start_dim=

0, end_dim=1)

.size(

)torch.size([6

,2])

class torch

.nn.flatten(start_dim=

1, end_dim=-1

)#flattens a contiguous range of dims into a tensor.

#for use with sequential. :

#param start_dim: first dim to flatten (default = 1).

#param end_dim: last dim to flatten (default = -1).

#能力有限,個人認為是用於卷積中的

#shape:

#input: (n, *dims)(n,∗dims)

#output: (n, \prod *dims)(n,∏∗dims) (for the default case).

#官方example

>>

> m = nn.sequential(

>>

> nn.conv2d(1,

32,5,

1,1)

,>>

> nn.flatten(

)>>

>

)#源**為 torch.nn.modules.flatten

from

.module import module

[docs]

class

flatten

(module)

: r"""

flattens a contiguous range of dims into a tensor. for use with :class:`~nn.sequential`.

args:

start_dim: first dim to flatten (default = 1).

end_dim: last dim to flatten (default = -1).

shape:

- input: :math:`(n, *dims)`

- output: :math:`(n, \prod *dims)` (for the default case).

examples::

>>> m = nn.sequential(

>>> nn.conv2d(1, 32, 5, 1, 1),

>>> nn.flatten()

>>> )

"""__constants__ =

['start_dim'

,'end_dim'

]def

__init__

(self, start_dim=

1, end_dim=-1

):super

(flatten, self)

.__init__(

) self.start_dim = start_dim

self.end_dim = end_dim

defforward

(self,

input):

return

input

.flatten(self.start_dim, self.end_dim)

和torch.flatten()一樣

openstack 官方文件閱讀之cells(二)

首次安裝 由於只有乙個api資料庫,因此它的連線資訊儲存在nova.conf檔案中。api database connection mysql pymysql root secretmysql dbserver nova api?charset utf8 由於可能存在多個 cell 資料庫 實際上每...

fedora chm 文件 閱讀

感冒了,疼痛的要命,沒法正常幹活,只能看看文件.沒辦法,裝乙個 1.yum 搜尋一下,看看fedora倉庫中有什麼是跟chm相關的 yum search chm 結果挺多,有幾項看著是 kchmviewer qt.x86 64 qt4 based chm viewer chm2pdf.noarch ...

Django文件閱讀之執行原始SQL查詢

django提供了兩種執行原始sql查詢的方法 可以使用manager.raw 來執行原始查詢並返回模型例項,或者可以完全避免模型層直接執行自定義sql。每次編寫原始sql時都要關注防止sql注入 一 raw 方法 raw 方法可以用來執行返回模型例項原始的sql查詢 manager.raw raw...