pytorch : srcIndex ＜ srcSelectDimSize

今天写pytorch代码，在调用Dropout时出现了pytorch dropout srcIndex < srcSelectDimSize错误，仔细看debug的位置，Dropout怎么也不应该出现这个问题。

想了半天，才反应过来，pytorch动态创建计算图，每个tensor操作或者函数调用时，都会触发所有前置的网络。

所以解决办法是，反向查找所有tensor的维度。

一个方便的办法是 print(model)，pytorch会输出模型的结构及每个tensor的维度。