PyTorch 中的正方形

請我喝杯咖啡☕

*備忘錄：

• 我的帖子解釋了 pow()。
• 我的貼文解釋了 float_power()。
• 我的帖子解釋了abs()和sqrt()。
• 我的帖子解釋了 gcd() 和 lcm()。
• 我的貼文解釋了trace()、reciprocal() 和rsqrt()。

square() 可以得到零個或多個元素平方的 0D 或多個 D 張量，得到零個或多個元素的 0D 或多個 D 張量如下所示：

*備忘錄：

• square() 可以與火炬或張量一起使用。
• 第一個參數（輸入）使用 torch 或使用張量（必需類型：int、float、complex 或 bool 的張量）。
• torch 存在 out 參數（可選-預設：無-型別：張量）： *備註：
  • 必須使用 out=。
  • 我的貼文解釋了論點。

import torch

my_tensor = torch.tensor(-3)

torch.square(input=my_tensor)
my_tensor.square()
# tensor(9)

my_tensor = torch.tensor([-3, 1, -2, 3, 5, -5, 0, -4])

torch.square(input=my_tensor)
# tensor([9, 1, 4, 9, 25, 25, 0, 16])

my_tensor = torch.tensor([[-3, 1, -2, 3],
                          [5, -5, 0, -4]])
torch.square(input=my_tensor)
# tensor([[9, 1, 4, 9],
#         [25, 25, 0, 16]])

my_tensor = torch.tensor([[[-3, 1], [-2, 3]],
                          [[5, -5], [0, -4]]])
torch.square(input=my_tensor)
# tensor([[[9, 1], [4, 9]],
#         [[25, 25], [0, 16]]])

my_tensor = torch.tensor([[[-3., 1.], [-2., 3.]],
                          [[5., -5.], [0., -4.]]])
torch.square(input=my_tensor)
# tensor([[[9., 1.], [4., 9.]],
#         [[25., 25.], [0., 16.]]])

my_tensor = torch.tensor([[[-3.+0.j, 1.+0.j], [-2.+0.j, 3.+0.j]],
                          [[5.+0.j, -5.+0.j], [0.+0.j, -4.+0.j]]])
torch.square(input=my_tensor)
# tensor([[[9.-0.j, 1.+0.j], [4.-0.j, 9.+0.j]],
#         [[25.+0.j, 25.-0.j], [0.+0.j, 16.-0.j]]])

my_tensor = torch.tensor([[[True, False], [True, False]],
                          [[False, True], [False, True]]])
torch.square(input=my_tensor)
# tensor([[[1, 0], [1, 0]],
#         [[0, 1], [0, 1]]])

以上是PyTorch 中的正方形的詳細內容。更多資訊請關注PHP中文網其他相關文章！