PyTorch基础-Tensor的属性,数据,运算-01

Tensor的属性

import torch

a = torch.tensor([1,2,3],dtype=int) # 创建一个整数tensor
print(a)
print(a.dtype)

b = torch.tensor([1,2,3],dtype=float)# 创建一个浮点数tensor
print(b)
print(b.dtype)

c = torch.tensor([[1,2,3],[4,5,6]])# # 创建一个二维tensor
print(c)
print(c.ndim)# 数据维度
print(c.shape) # 数据形状
print(c.dtype) # 数据类型

Tensor的数据创建

import torch

torch.ones(2,3) # 创建一个2行3列全为1的数据

torch.zeros(3,3) # 创建一个3行3列全为0的数据

torch.rand(3,4)# 生成一个3行4列的随机数（0-1之间）

torch.randint(0,10,(2,3))# 生成一个0-10之间的2行3列的整数