//开启梯度训练
(loss), grads = grad_fn(mini_batch)

//报错位置
---> 84 deg_inv[deg_inv == float('inf')] = 0 
//报错原因
   2697 def __call__(self, data, index, value):
-> 2698     return Tensor_.setitem_index_info(data, index, value, self.is_ascend)
RuntimeError: Output device address's size 1 is not equal output_size_list's size 2
----------------------------------------------------
- C++ Call Stack: (For framework developers)
----------------------------------------------------
mindspore/ccsrc/runtime/pynative/run_op_helper.cc:644 UpdateOutputDeviceInfo

//报错位置
File "/root/mbe-mindspore/mindspore_src/utils/opsi.py", line 165, in tile_along_beam
for i, d in enumerate(v.shape):
File "/root/miniconda3/lib/python3.8/site-packages/mindspore/common/_stub_tensor.py", line 85, in shape
self.stub_shape = self.stub.get_shape()
//报错信息
RuntimeError: Output device address's size 1 is not equal output_size_list's size 2
----------------------------------------------------
- C++ Call Stack: (For framework developers)
----------------------------------------------------
mindspore/ccsrc/runtime/pynative/run_op_helper.cc:644 UpdateOutputDeviceInfo