PyTorch_创建线性和随机张量

1. torch.arange 和 torch.linspace     创建线性张量
2. torch.random.init_seed 和 torch.random.manual_seed     随机种子设置
3. torch.randn     创建随机张量

代码

import torch 
import numpy as np # 创建线性张量
def test01():# 创建指定步长的张量data = torch.arange(0, 10, 2)  # (开始值，结束值，步长)print(data)# 在指定区间指定元素个数data = torch.linspace(0, 10, 1000)  # (开始值，结束值，元素个数)print(data)# 创建随机张量
def test02():# 创建随机张量data = torch.randn(2, 3)print(data)# 固定随机数种子torch.random.manual_seed(0)print(torch.random.initial_seed()) # 获取当前随机数种子if __name__ == "__main__":test02()