torch.sum(temp_target[...,:-1] * temp_inputs, axis=[0,1])
时间: 2024-06-06 13:09:15 浏览: 178
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
这是一个求和操作,其中temp_target是目标序列,temp_inputs是输入序列。[...,:-1]是对目标序列进行索引,取出除了最后一个元素以外的所有元素。*表示逐个元素相乘,axis=[0,1]表示对第0维和第1维进行求和。最终结果是一个标量,表示目标序列和输入序列对应位置元素相乘后的和。
阅读全文