本项目采用PyTorch框架实现了一系列先进的注意力机制模型,旨在提升深度学习模型在序列数据处理中的性能与效率。
import math
import torch
import torch.nn as nn
import os
def file_name_walk(file_dir):
for root, dirs, files in os.walk(file_dir):
print(root) # 当前目录路径
print(dirs) # 当前路径下所有子目录
print(files) # 当前路径下所有非目录子文件
file_name_walk(/home/kesci/input)