torch的数据基本格式（李沐老师课程）

「已注销」

已于 2024-08-28 10:57:31 修改

阅读量357

点赞数 5

文章标签： python 深度学习人工智能

于 2024-08-28 10:09:27 首次发布

本文链接：https://blog.youkuaiyun.com/zhujiale5200/article/details/141633664

版权

"""
张量的运行
"""
import torch

dataZ = torch.arange(12)
print("dataZ", dataZ)

print("data的形状", dataZ.shape)

print("元素的总数", dataZ.numel())

"""
改变元素形状
"""
print(dataZ.reshape(3, 4))

"""
全0的张量
"""
print(torch.zeros(3, 4))

"""
全1的张量
"""
print(torch.ones(3, 4))

"""
特定的张量
"""
print(torch.tensor([[1, 2, 3, 4], [2, 2, 3, 4]]))