torch.ones和torch.new_ones和torch.ones_like區別和聯系


官方文檔解釋:
To create a tensor with pre-existing data, use torch.tensor().

To create a tensor with specific size, use torch.* tensor creation ops (see Creation Ops).

To create a tensor with the same size (and similar types) as another tensor, use torch.*_like tensor creation ops (see Creation Ops).

To create a tensor with similar type but different size as another tensor, use tensor.new_* creation ops.
簡單來說,
torch.tensor() 用的最多,一般都用它;
torch.* 用於創建特殊形式的 tensor,包括 torch.ones()torch.zeros()等;
torch.*_like() 用於創建一個與已知 tensor 形狀相同的 tensor;
torch.new_* 用於創建一個與已知 tensor 類型相同的 tensor
最后放一下參數表
torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False) → Tensor
torch.ones(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
torch.ones_like(input, dtype=None, layout=None, device=None, requires_grad=False) → Tensor
torch.new_ones(size, dtype=None, device=None, requires_grad=False) → Tensor


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM