PyTorch

Windows 下 Visual Studio + Libtorch 环境搭建

2024-12-19
软件准备 Visual Studio 下载地址 OpenCV 可直接下载预编译版本 下载地址 下载后选择纯英文解压路径,后续以 D:\opencv 示例 Libtorch 根据实际情况选择下载版本 下载地址 下载完成后解压,后...
Read more

爱因斯坦求和约定 torch.einsum()

2022-05-23
使用 PyTorch 进行矩阵运算时,总是需要去记忆形如 torch.dot, torch.mm 的函数,当张量维度上升到四维及以上时,torch.einsum 是一种能够表达矩阵点积、外积、转置等运算,包括部分复杂张量运算在内的优雅方式。...
Read more
爱因斯坦求和约定 torch.einsum()