Lines Matching full:allclose
142 self.assertTrue(torch.allclose(C.order(i, j), torch.mm(A_, B_)))
143 self.assertTrue(torch.allclose(torch.triu(A_, 0), triu(A_)))
184 torch.allclose(a_out, b_out)
218 self.assertTrue(torch.allclose(a_out, b_out))
267 self.assertTrue(torch.allclose(a_out, b_out))
315 # self.assertTrue(torch.allclose(a.order(i, j), i.expand(j).order(i, j)))
316 # self.assertTrue(torch.allclose(b.order(i, j), j.expand(i).order(i, j)))
323 self.assertTrue(torch.allclose(r.order(i, j), ap.max(2)[0]))
343 self.assertTrue(torch.allclose(a, x))
352 assert torch.allclose(r, A @ B)
357 assert torch.allclose(A.sum(), A[i].sum(0, keepdim=True).sum((0, i)))
358 assert torch.allclose(A[i].std(i, True), A.std(0, True))
360 assert torch.allclose(A[i, k].max(i)[0].order(k), A.max(0)[0])
361 assert torch.allclose(A.sort(1)[0], A[i, k].sort(k)[0].order(i, k))
363 # assert torch.allclose(A.chunk(2,1)[0], A[i, k].chunk(2, k)[0].order(i, k))
364 assert torch.allclose(A[i].renorm(1, i, 7).order(i), A.renorm(1, 0, 7))
366 … # assert torch.allclose( torch.stack([A, A], 1), stack([A[i,k], A[i, k]], kk, k).order(i, kk, k))
370 # assert torch.allclose(torch.cat([A, A], 1), r.order(i, k2))
373 assert torch.allclose(A.expand(5, -1, -1), A[i, k].expand(j).order(j, i, k))
376 assert torch.allclose(A[:, 0:2], A[i, k].index(k, C[z]).order(i, z))
381 assert torch.allclose(r.order(i, o, l), A.view(-1, 2, 2))
383 assert torch.allclose(A, rr.order(i, k))
387 assert torch.allclose(r.order(i, k), r2)
390 assert torch.allclose(A.T, A[..., k].order(k))
394 assert torch.allclose(A[i, a_].order(*a_, i), A.T)
396 assert torch.allclose(A[:, a_].order(*a_), A.T)
398 assert torch.allclose(A[i, b_, k].order(i, k, *b_), A)
401 assert torch.allclose((A[i] + i).order(i), A + torch.arange(3)[:, None])
410 assert torch.allclose(A[i, [c, d]].order(i, c, d), A.view(3, 2, 2))
412 assert torch.allclose(
429 assert torch.allclose(a, b.order(s, d))
432 assert torch.allclose(
454 assert torch.allclose(C, A @ B)
488 assert torch.allclose(r1.order(i, j), r0)
542 assert torch.allclose(r2, r, atol=1e-06)
581 assert torch.allclose(f / l, torch.nn.functional.softmax(a, dim=0))
591 assert torch.allclose(r.order(i, o, l), A.view(-1, 2, 2))
593 assert torch.allclose(A, rr.order(i, k))
597 assert torch.allclose(A[:, 0:2], x)
601 assert torch.allclose(
629 assert torch.allclose(A.permute(1, 3, 0, 2), r)
634 assert torch.allclose(A[i].order(1, i), A.permute(2, 0, 1))
679 assert torch.allclose(values, values_)