溫馨提示×

Python中torch.clamp()函數怎么使用

小億
346
2024-01-22 13:00:07
欄目: 編程語言

torch.clamp()函數用于將輸入張量中的元素限制在指定的范圍內。

函數的語法如下:

torch.clamp(input, min, max, out=None)

參數說明:

  • input:輸入的張量。
  • min:最小值,小于該值的元素將被替換為該值。
  • max:最大值,大于該值的元素將被替換為該值。
  • out:用于存儲輸出結果的張量。

使用示例:

import torch

x = torch.randn(5)  # 創建一個包含5個隨機數的張量
print(x)  # 打印原始張量

# 使用clamp函數將張量的元素限制在-0.5到0.5之間
y = torch.clamp(x, -0.5, 0.5)
print(y)  # 打印限制后的張量

輸出結果示例:

tensor([ 0.0849, -0.2706,  0.7244,  0.0921,  0.6237])
tensor([ 0.0849, -0.2706,  0.5000,  0.0921,  0.5000])

在上面的示例中,首先創建了一個包含5個隨機數的張量x,然后使用clamp函數將該張量的元素限制在-0.5到0.5之間,結果存儲在張量y中。最后打印出原始張量和限制后的張量??梢钥吹?,張量y中的元素都被限制在了-0.5到0.5之間。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女