[()] [tensor([])] [tensor([[0., 1.], [1., 0.], [2., 0.], [2., 2.]])] [()] [()] [()] [tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99])] [tensor([-1.-1.j, -2.-2.j, 3.+3.j])] [tensor([0, 0, 0])] [()] [()] [tensor([[6, 1], [7, 2], [8, 3]])] [tensor([[[0.0000e+00, 0.0000e+00, 3.0520e-34], [0.0000e+00, 1.6874e-35, 4.5916e-41]]])] [tensor([[0., 0., 0.], [0., 0., 0.]])] [tensor([50.0000, 51.3158, 52.6316, 53.9474, 55.2632, 56.5789, 57.8947, 59.2105, 60.5263, 61.8421, 63.1579, 64.4737, 65.7895, 67.1053, 68.4211, 69.7368, 71.0526, 72.3684, 73.6842, 75.0000])] [] [()] [()] [tensor(0.3330), tensor(0.4105)] [tensor([[0.9689, 0.5078, 0.4918], [0.7814, 0.6311, 0.3206], [0.8915, 0.5919, 0.1167]]), tensor([[0, 1, 2], [2, 3, 0], [0, 3, 1]])] [tensor(0.4687)] [tensor([[0.3584], [0.4163], [0.4067], [0.6933]])] [tensor(0.0036), tensor(0.9689)] [tensor([[5., 5.], [5., 5.]])] [tensor([[-3., -3.], [-3., -3.]])] [tensor([[1.3758, 1.1868, 0.9660, 0.5027, 0.4559], [1.5032, 1.4792, 0.8997, 0.3444, 0.5422], [0.9266, 1.0152, 1.0149, 0.3135, 0.6292]])] [torch.float32] [()] [tensor([0.7881, 0.9417, 0.1911, 0.0879]), tensor([0.8846, 0.4991, 0.4789, 0.8133]), tensor([0.1656, 0.8934, 0.6181, 0.4339]), tensor([0.1984, 0.0921, 0.9596, 0.3735])] [tensor([[0.7881, 0.9417], [0.8846, 0.4991], [0.1656, 0.8934], [0.1984, 0.0921]]), tensor([[0.1911, 0.0879], [0.4789, 0.8133], [0.6181, 0.4339], [0.9596, 0.3735]])] [()] [tensor([[0., 1., 0.], [1., 0., 0.], [1., 0., 1.]]), (tensor([0.0000, 0.6667, 1.3333, 2.0000]), tensor([0.0000, 0.6667, 1.3333, 2.0000]))] [()] [tensor([[1, 2], [1, 3], [2, 3]])] [False] [False] [[tensor([[0.9689, 0.5078, 0.4918, 0.2002], [0.3206, 0.2008, 0.7814, 0.6311], [0.8915, 0.1167, 0.0036, 0.5919]], requires_grad=True)]] [[tensor([[0.3824, 0.5299, 0.5226, 0.1151, 0.2418], [0.7855, 0.0685, 0.4314, 0.6370, 0.1590], [0.8955, 0.9329, 0.1465, 0.0437, 0.0148], [0.8291, 0.8978, 0.8416, 0.2305, 0.6674]], requires_grad=True)]] [()] [True] [()] [True] [[None]] [()] [FooModel( (linear): Linear(in_features=2, out_features=2, bias=True) (softmax): Softmax(dim=1) )] [tensor([[1.3758, 1.1868, 0.9660, 0.5027, 0.4559], [1.5032, 1.4792, 0.8997, 0.3444, 0.5422], [0.9266, 1.0152, 1.0149, 0.3135, 0.6292]], grad_fn=)] [()] [tensor([ 0., 1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 11., 12., 13., 14., 15., 16., 17., 18., 19., 20., 21., 22., 23., 24., 25., 26., 27., 28., 29., 30., 31., 32., 33., 34., 35., 36., 37., 38., 39., 40., 41., 42., 43., 44., 45., 46., 47., 48., 49., 50., 51., 52., 53., 54., 55., 56., 57., 58., 59., 60., 61., 62., 63., 64., 65., 66., 67., 68., 69., 70., 71., 72., 73., 74., 75., 76., 77., 78., 79., 80., 81., 82., 83., 84., 85., 86., 87., 88., 89., 90., 91., 92., 93., 94., 95., 96., 97., 98., 99.], requires_grad=True)] [()] [] [()] [()] [()] [()] [()] [()] [NeuralNetwork( (flatten): Flatten(start_dim=1, end_dim=-1) (linear_relu_stack): Sequential( (0): Linear(in_features=784, out_features=512, bias=True) (1): ReLU() (2): Linear(in_features=512, out_features=512, bias=True) (3): ReLU() (4): Linear(in_features=512, out_features=10, bias=True) ) )] [()] [()] [()] [()]