Thanks! I'm trying to optimize my custom neural net implementation in pytorch. Do you happen to know which of the two functions would be a more efficient implementation of the two operation? I'm curious to know if one function is faster than the other.