簡體   English   中英

如何將 PyTorch 張量的所有元素乘以常數?

[英]How do I multiply all elements of a PyTorch tensor by a constant?

假設我想將一個張量乘以一個常數(如 2),張量定義如下:

import torch
target = torch.tensor([ [1.5, 2.5], [.5, 1.0], ... ])
>> target
tensor([ [1.5, 2.5], [.5, 1.0], ... ])

我如何將張量的每個元素相乘以得出以下結果:

>>> target
tensor( [ 3.0, 5.0], [1.0, 2.0], ... ] ])

我在 pytorch 網站上找不到任何表明支持這樣的操作的內容,所以我的想法是將張量轉換為 numpy 數組,然后將該數組乘以 2,然后再轉換回 Z1AFA0EDDD6A79D80CF860A81。 是否有一種不那么迂回的方法來完成上述操作?

只需使用out = 2 * tensor

補充: tensor * tensor怎么做

output = tensor1.mul(tensor2)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM