在Torch中,可以使用torch.tensor()
函数来创建一个张量。例如:
import torch
# 创建一个大小为3x3的随机张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(tensor)
这样就创建了一个3x3的随机张量。也可以使用torch.zeros()
、torch.ones()
等函数来创建全0或全1的张量。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>