最近因為工作關係，主管請了講師來幫大家上課，其中講師介紹MSE 損失函數時，有個同事好奇問可不可以自訂損失函數，並且在裡面加入for loop呢？
講師回答，就他所知自訂損失函數可以，但要在損失函數裡面用迴圈不可行。
請問各位高手真的沒辦法嗎？謝謝

如果不能微分就不能用反向傳播

現在的框架的computational graph應該都支援條件判斷跟for loop 當然就跟樓上的講的一樣不能微分就不行但有例外 例如絕對值在pytorch是可以拿到微分資訊的分條件判斷

tf.vectorized_map 用這個代替？

不如說說你想用for loop算什麼