sqrt (), but all PyTorch math functions. I imagine it's easier to have torch.sqrt() accept Python scalars then to automatically vectorize math.sqrt() (a la ...
確定! 回上一頁