*位置乘 符号*在pytorch中是按位置相乘,存在广播机制。 例子: vec1 = torch.arange(4) vec2 = torch.tensor([4,3,2,1]) mat1 = torch.arange(12).reshape(4,3) mat2 = torch.arange(12).reshape(3,4) print(vec1 * vec2) print(mat2 * vec1) print(mat1 * mat1) Output: tensor([
专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]
Copyright (C)ICode9.com, All Rights Reserved.
ICode9版权所有