溫馨提示×

pytorch怎么復制一個張量

小億
139
2024-04-17 16:11:07
欄目: 深度學習

在PyTorch中，可以使用clone()函數(shù)來復制一個張量。例如：

import torch

# 創(chuàng)建一個張量
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 復制張量
y = x.clone()

print(y)

上面的代碼中，首先創(chuàng)建了一個張量x，然后使用clone()函數(shù)來復制這個張量，并將復制的張量賦值給y。最終打印出y的值也將會是與x相同的值。

0 踩