torch.flatten works in a counterintuitive way when supplied with 0-dim tensors. In numpy: np.asfarray(123).flatten() array([123.]) ...
確定! 回上一頁