爱因斯坦求和(enisum)
2024-09-27
在pytorch中的点积/外积/转置/矩阵乘法都是可以用einsum记法来表示的,包括复杂张量运算在内的优雅方式。会熟练运用einsum后,可以不用看库函数,还可以更加迅速的编写紧凑,高效的代码,可忽略不必要的张良转置和变形,甚至还可以忽略生成中间张量。事实上,pytorch引入的能够自动生成GPU代码并且为特定输入尺寸自动调节的张量理解就类似于einsum领域的特定语言。
645 字
|
3 分钟