Lines Matching full:rand

15 def rand(*shape):  function
16     return torch.rand(*shape).mul(16).add(1)
23     return (rand(1), rand(1))
27     return (rand(32), rand(32))
31     return (rand(1, 32), rand(1, 32))
35     return (rand(4, 32), rand(32))
39     return (rand(32, 12, 64, 64), rand(32, 12, 64, 64))
43     return (rand(32, 12, 64, 64)[..., ::2], rand(32, 12, 64, 64)[..., ::2])
48         rand(32, 12, 64, 64).transpose(-1, -2),
49         rand(32, 12, 64, 64).transpose(-1, -2),
54     return (rand(32, 3, 224, 224), rand(32, 3, 224, 224))
58     return (rand(16, 32, 64), rand(16, 32, 64))
63         rand(32, 3, 224, 224).to(memory_format=torch.channels_last),
64         rand(32, 3, 224, 224).to(memory_format=torch.channels_last),
69     return (rand(32, 12, 64, 64), rand(64))
73     return (rand(32, 3, 223, 223).to(memory_format=torch.channels_last), rand(3, 1, 1))
77     return (rand(8192, 8192), rand(8192, 8192))
81     return (rand(8192, 8192).transpose(0, 1), rand(8192, 8192).transpose(0, 1))
86         rand(32, 32, 256, 256).to(memory_format=torch.channels_last),
87         rand(32, 32, 256, 256).to(memory_format=torch.channels_last),
92     return (rand(1, 32, 32, 2), rand(1024, 1, 1, 2))
96     return (rand(64, 8, 256, 162), rand(256, 162))