[英]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.