a=np.array([[1,2],[3,4]]) a[...,None].shape returns: (2, 2, 1) However: ta=torch.from_numpy(a) ta[...,None].size() returns: (2, 1, ...
確定! 回上一頁