torch.spmm矩阵乘法
矩阵 乘法 torch
2023-09-14 09:08:53 时间
Example:
import torch
indices = torch.tensor([[0,1],
[0,1]])
values = torch.tensor([2,3])
shape = torch.Size((2,2))
s = torch.sparse.FloatTensor(indices,values,shape)
print(s)
d = torch.tensor([[1,2],
[3,4]])
print(d)
print(torch.spmm(s,d))
"""
tensor(indices=tensor([[0, 1],
[0, 1]]),
values=tensor([2, 3]),
size=(2, 2), nnz=2, layout=torch.sparse_coo)
tensor([[1, 2],
[3, 4]])
tensor([[ 2, 4],
[ 9, 12]])
"""
相关文章
- Java实现 蓝桥杯 算法提高 矩阵乘法(暴力)
- Java实现 蓝桥杯 算法提高 矩阵乘法(暴力)
- (算法)最大子数组和以及最大子矩阵和
- (剑指Offer)面试题20:顺时针打印矩阵
- LeetCode-1139. 最大的以 1 为边界的正方形【前缀和,矩阵】
- 【STM32F429的DSP教程】第22章 DSP矩阵运算-放缩,乘法和转置矩阵
- Java里的稀疏矩阵Sparse Array
- 【[Offer收割]编程练习赛13 D】骑士游历(矩阵模板,乘法,加法,乘方)
- 不同维度矩阵乘法的试验
- 矩阵微积分
- MATLAB | 好看的相关系数矩阵图绘制
- 2.3CUDA矩阵乘法
- 27 | SIMD:如何加速矩阵乘法?
- AI学习之路(19)TensorFlow里的矩阵乘法
- torch中向量、矩阵乘法大总结
- 【机器学习】前置知识:矩阵的表示与定义 | Identity 身份矩阵 | 逆矩阵和转置 | 标量乘法
- 矩阵乘法脉动阵列的C++模拟
- EIE稀疏矩阵乘法硬件模拟