pytorch 三维以内的张量乘法
'''tensor乘以标量数值'''
y=torch.mul(x,value)
#或者
y=x.mul(value)
'''一维张量乘以一维张量'''
# x.shape=torch([m]).y.shape=torch([m]),z.shape=torch(value)
z=torch.dot(x,y)
# 或者
z=torch.randn(y,x)
'''二维张量相乘'''
# x.shape=torch([m,n]),y.shape=torch([n,d]),z.shape=torch([
原创
2020-12-15 14:44:20 ·
2827 阅读 ·
0 评论