首頁 > 後端開發 > Python教學 > PyTorch 中的任何一個

PyTorch 中的任何一個

Mary-Kate Olsen
發布: 2025-01-01 09:14:11
原創
497 人瀏覽過

any in PyTorch

請我喝杯咖啡☕

*我的貼文解釋了所有()。

any() 可以檢查 0D 或更多 D 張量的任何元素是否為 True,得到零個或更多元素的 0D 或更多 D 張量,如下所示:

*備忘錄:

  • any() 可以與 torch 或張量一起使用。
  • 第一個參數(輸入)使用 torch 或使用張量(必要類型:int、float、complex 或 bool 的張量)。
  • 帶有 torch 的第二個參數或帶有張量的第一個參數是暗淡的(可選類型:int、int 元組或 int 列表)。
  • 帶有 torch 的第三個參數或帶有張量的第二個參數是 keepdim(Optional-Default:False-Type:bool)。 *我的貼文解釋了 keepdim 的論點。
  • torch 存在 out 參數(可選-預設:無-型別:張量): *備註:
    • 必須使用 out=。
    • 我的貼文解釋了論點。
  • 空張量回傳 1D 或更多 D 張量或空的 1D 或更多 D 張量的 False。
import torch

my_tensor = torch.tensor(True)

torch.any(input=my_tensor)
my_tensor.any()
torch.any(input=my_tensor, dim=0)
torch.any(input=my_tensor, dim=-1)
torch.any(input=my_tensor, dim=(0,))
torch.any(input=my_tensor, dim=(-1,))
# tensor(True)

torch.any(input=my_tensor, dim=0, keepdim=True)
# tensor(True)

my_tensor = torch.tensor([True, False, True, False])

torch.any(input=my_tensor)
torch.any(input=my_tensor, dim=0)
torch.any(input=my_tensor, dim=-1)
torch.any(input=my_tensor, dim=(0,))
torch.any(input=my_tensor, dim=(-1,))
# tensor(True)

torch.any(input=my_tensor, dim=0, keepdim=True)
# tensor([True])

my_tensor = torch.tensor([[True, False, True, False],
                          [True, False, True, False]])
torch.any(input=my_tensor)
torch.any(input=my_tensor, dim=(0, 1))
torch.any(input=my_tensor, dim=(0, -1))
torch.any(input=my_tensor, dim=(1, 0))
torch.any(input=my_tensor, dim=(1, -2))
torch.any(input=my_tensor, dim=(-1, 0))
torch.any(input=my_tensor, dim=(-1, -2))
torch.any(input=my_tensor, dim=(-2, 1))
torch.any(input=my_tensor, dim=(-2, -1))
# tensor(True)

torch.any(input=my_tensor, dim=0)
torch.any(input=my_tensor, dim=(0,))
torch.any(input=my_tensor, dim=-2)
# tensor([True, False, True, False])

torch.any(input=my_tensor, dim=1)
torch.any(input=my_tensor, dim=-1)
torch.any(input=my_tensor, dim=(-1,))
# tensor([True, True])

torch.any(input=my_tensor, dim=0, keepdim=True)
# tensor([[True, False, True, False]])

my_tensor = torch.tensor([[0, 1, 2, 3],
                          [4, 5, 6, 7]])
torch.any(input=my_tensor)
# tensor(True)

my_tensor = torch.tensor([[0., 1., 2., 3.],
                          [4., 5., 6., 7.]])
torch.any(input=my_tensor)
# tensor(True)

my_tensor = torch.tensor([[0.+0.j, 1.+0.j, 2.+0.j, 3.+0.j],
                          [4.+0.j, 5.+0.j, 6.+0.j, 7.+0.j]])
torch.any(input=my_tensor)
# tensor(True)

my_tensor = torch.tensor([[]])

torch.any(input=my_tensor)
# tensor(False)

torch.any(input=my_tensor, dim=0)
torch.any(input=my_tensor, dim=-2)
# tensor([], dtype=torch.bool)

torch.any(input=my_tensor, dim=1)
torch.any(input=my_tensor, dim=-1)
# tensor([False])
登入後複製

以上是PyTorch 中的任何一個的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:dev.to
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板