【Pytorch】常用操作

1 os，文件相关操作

1.1 返回文件夹下所有文件

• os.listdir：返回文件夹下的所有子文件名称

for file_name in os.listdir(file_path):

• os.walk：返回文件夹下的所有子文件、及子文件内的所有文件：绝对路径、子文件夹、文件名。

for root, dirs, files in os.walk(path):

1.2 文件夹下文件个数

n = len(os.listdir(file_path))

1.3 文件夹目录和文件名合并

os.path.join(sph_path, sph_list[idx])

2 HDF5文件的保存和读取

HDF5文件

3 音频文件的保存和读取

librosa.load

100 创建随机tensor并保存为mat

创建1-255,16*16的tensor
mat中的变量名为result，文件名为save.mat

tensor = torch.randint(1,255,[16,16]) 
tensor = tensor.cpu().detach().numpy()
result = np.array(tensor)
io.savemat('save.mat', {'result':result})