2022年7月24日
torch中乘法整理,*&torch.mul()&torch.mv()&torch.mm()&torch.dot()&@&torch.mutmal()
摘要： *位置乘 符号*在pytorch中是按位置相乘，存在广播机制。 例子： vec1 = torch.arange(4) vec2 = torch.tensor([4,3,2,1]) mat1 = torch.arange(12).reshape(4,3) mat2 = torch.arange(12) 阅读全文
posted @ 2022-07-24 20:53 CharlesLC 阅读(798) 评论(0) 推荐(0)