In [1]: import torch import numpy as np In [2]: a ... False], [False, False]]) In [14]: torch.sqrt(a) Out[14]: tensor([[1.0000, 1.4142], ...
確定! 回上一頁