PyTorch :每一维中遍历所需要经过的元素个数,返回类型为元组。 ... import torch as tc b = tc.tensor([[1, 2, 3], [7, 8, 9]]) print(b.stride()) # out: # (3, 1) ...
確定! 回上一頁