定义tensor数组时使用dtype=torch.float , 统一成float格式
即可矩阵内积(类似 np.dot(x, W) )
y = torch.matmul(x, W)
pytorch中实现与np.dot类似功能
于 2022-03-18 11:23:49 首次发布
定义tensor数组时使用dtype=torch.float , 统一成float格式
即可矩阵内积(类似 np.dot(x, W) )
y = torch.matmul(x, W)