PyTorch

使用 PyTorch 进行矩阵运算时，总是需要去记忆形如 torch.dot, torch.mm 的函数，当张量维度上升到四维及以上时，torch.einsum 是一种能够表达矩阵点积、外积、转置等运算，包括部分复杂张量运算在内的优雅方式。...