Torch中怎么创建一个张量

在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的张量。

版权声明

本文内容由互联网用户自发贡献，该文观点仅代表作者本人
本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。