dtype – A str, Python numeric or string type, NumPy data type, or PyTorch dtype. ... x = torch.ones(d_1, d_2, d_3, d_4) y = flatten(x, 2) # y.shape == [d_1, ...
確定! 回上一頁