![](https://img-blog.csdnimg.cn/d361ed8280f248e8acdff4e0b46fbbb8.jpeg?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch
文章平均质量分 51
笔记
Osatnb.c
计算机理工男,欢迎学习交流
展开
-
torch.bmm函数讲解
例如,假设我们有两个批次的矩阵 A 和 B,维度分别为 (2, 3, 4) 和 (2, 4, 5)。我们可以使用 torch.bmm 将它们相乘。torch.bmm 是 PyTorch 中的一个函数,用于执行批矩阵乘法(batch matrix multiplication)操作。torch.bmm 将批中的每对矩阵相乘,返回一个新的三维张量,形状为 (batch_size, n, p)。其中 n 是第一个矩阵的列数,m 是两个矩阵共享的维度,p 是第二个矩阵的列数。原创 2023-10-01 17:44:53 · 5549 阅读 · 0 评论 -
Pytorch visdom知识(看这篇文章就够了!)(代码+注释+图示)
超级详细原创 2023-08-11 17:38:06 · 371 阅读 · 1 评论