矩阵乘法pytorch._c

最新推荐文章于 2025-03-13 23:55:42 发布

原创

最新推荐文章于 2025-03-13 23:55:42 发布 · 300 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #抽象代数

本文深入探讨PyTorch库中的_C模块，重点解析如何使用该模块进行高效矩阵乘法操作，结合抽象代数概念，阐述底层计算原理。

在这里插入图片描述

In [1]: import torch

In [2]: x=torch.randn(2)

In [3]: w=torch.randn(2,3)

In [4]: x@w
Out[4]: tensor([0.9364, 1.3934, 0.3611])

In [5]: _.shape
Out[5]: torch.Size([3])

In [8]: x.repeat(2,1)
Out[8]:
tensor([[ 0.2738, -1.1417],
        [ 0.2738, -1.1417]])

In [9]: x.repeat(2