self.assertAllClose(out[0][0][0], out[1][0][3])