矩阵乘法pytorch._c

codingDL

于 2020-06-23 20:03:15 发布

阅读量284

点赞数

CC 4.0 BY-SA版权

文章标签： pytorch 抽象代数

本文链接：https://blog.youkuaiyun.com/foreachself/article/details/106930993

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

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