2020-11-22

RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 3 and 2  报错

错误代码如下：

注意，[ ... ,10] != [... , 10:11] 左边是二维的，右边是三维的，维数更高。这种错误自己犯了好多次了，要养成时刻注意维数对齐的习惯。

正确代码如下：