溫馨提示×

PyTorch創(chuàng)建張量的方法是什么

小億
83
2024-05-10 15:29:52

在PyTorch中，可以使用torch.Tensor()函數(shù)或者torch.tensor()函數(shù)來創(chuàng)建張量。示例如下：

import torch

# 使用torch.Tensor()函數(shù)創(chuàng)建一個空的張量
empty_tensor = torch.Tensor()

# 使用torch.tensor()函數(shù)創(chuàng)建一個具有指定數(shù)據(jù)的張量
data = [[1, 2, 3], [4, 5, 6]]
tensor_data = torch.tensor(data)

print(empty_tensor)
print(tensor_data)

此外，還可以使用torch.zeros()、torch.ones()、torch.randn()等函數(shù)創(chuàng)建特定形狀和類型的張量。

0 踩