首先,创建用于写入的文件(这里举例Ubuntu系统)
touch graphq.np
保存tensor
gpath_temp = '/root/KING/graph_features' # 文件存储目录
file_name = os.path.join(gpath_temp, 'graphq.np') # 写入graphq.np文件
g = recon_s2.cuda().data.cpu().numpy()
np.savetxt(file_name, g)