torch.all(tens_a.eq(tens_b)) ... Or torch.equal for the whole tensor exactly: ... For that kind of comparison you have torch.allclose .
確定! 回上一頁