您現在的位置是：網站首頁>JAVApytorch常用函數之torch.randn()解讀

pytorch常用函數之torch.randn()解讀

宸宸2024-05-26【JAVA】68人已圍觀

給網友們整理相關的編程文章，網友崔蕭玉根據主題投稿了本篇教程內容，涉及到pytorch常用函數、pytorch torch.randn()、pytorch函數、pytorch函數之torch.randn()相關內容，已被763網友關注，下麪的電子資料對本篇知識點有更加詳盡的解釋。

pytorch函數之torch.randn()

pytorch常用函數torch.randn()

torch.randn(*sizes, out=None) → Tensor

功能：從標準正態分佈（均值爲0，方差爲1）中抽取的一組隨機數。返廻一個張量

sizes (int…) - 整數序列，定義輸出張量的形狀
out (Tensor, optinal) - 結果張量

eg：

random = torch.randn(2, 3）
out:  0.5419 0.1594 -0.0413
        -2.7937 0.9534 0.4561

pytorch torch.chunk(tensor, chunks, dim)

說明：在給定的維度上講張量進行分塊。

蓡數：

tensor(Tensor) -- 待分塊的輸入張量
chunks(int) -- 分塊的個數
dim(int) -- 維度，沿著此維度進行分塊

>>> x = torch.randn(3, 3)
>>> x
tensor([[ 1.0103,  2.3358, -1.9236],
        [-0.3890,  0.6594,  0.6664],
        [ 0.5240, -1.4193,  0.1681]])
>>> torch.chunk(x, 3, dim=0)
(tensor([[ 1.0103,  2.3358, -1.9236]]), tensor([[-0.3890,  0.6594,  0.6664]]), tensor([[ 0.5240, -1.4193,  0.1681]]))
>>> torch.chunk(x, 3, dim=1)
(tensor([[ 1.0103],
        [-0.3890],
        [ 0.5240]]), tensor([[ 2.3358],
        [ 0.6594],
        [-1.4193]]), tensor([[-1.9236],
        [ 0.6664],
        [ 0.1681]]))
>>> torch.chunk(x, 2, dim=1)
(tensor([[ 1.0103,  2.3358],
        [-0.3890,  0.6594],
        [ 0.5240, -1.4193]]), tensor([[-1.9236],
        [ 0.6664],
        [ 0.1681]]))