墨瞳的Blog

torch.manual_seed(2020) # cpu的随机数种子torch.cuda.manual_seed() # 当前gpu的随机数种子torch.cuda.manual_seed_all() # 所有gpu的随机数种子创建tensora = torch.tensor([1, 2, 3])print(a) # tensor([1, 2, 3])b = torch.empty(1, 3) # 创建一个指定大小但未初始化的tensorprint(b

python文件操作，创建读写保存文件文件夹，提取修改文件名

背景为了匹配下一个环节的输入，需要将手头已有的 n x 3 维的 txt文件转换为 1024 x 3维的数据，n可能比1024大也可能比1024小。大于1024维的数据进行随机采样，小于1024维的数据先复制内容进行扩充到大于1024，然后降维。代码：#!/usr/bin/env python# -*- coding: utf-8 -*-import json, os, reimport randomimport numpy as npsource_file = 'part0.txt'r

代码：#!/usr/bin/env python# -*- coding: utf-8 -*-import json,os,re source_file='part1.csv'json_file_result='part1.json'zd_key = ["name", "size", "container", "volume"]# 判断文件是否存在，并清空文件内容if os.path.exists(json_file_result): with open(json_file_result