Пример #1
0
    def makeTensor(self, name):
        """ Create a tensor from the current interactively set attributes """ 

        rank_ids = self.rank_ids[name]

        t = Tensor.fromRandom(name=name,
                              rank_ids=rank_ids,
                              shape=[self.variables[r+"_SHAPE"] for r in rank_ids],
                              density=(len(rank_ids)-1)*[1.0]+[self.variables[name+"_DENSITY"]],
                              interval=self.variables[name+"_INTERVAL"],
                              seed=self.variables[name+"_SEED"],
                              color=self.variables[name+"_COLOR"])

        return t
Пример #2
0
    def test_fromRandom_wo_ids(self):
        """Test construction of a random tensor without rankids"""

        rank_ids = ["R1", "R0"]
        shape = [10, 10]
        tensor_ref = Tensor.fromUncompressed(
            rank_ids,
            [[0, 10, 10, 1, 0, 9, 8, 0, 0, 3], [
                9, 1, 0, 10, 1, 0, 10, 0, 0, 0
            ], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 3, 0, 3, 5, 0, 5, 7, 0, 0],
             [6, 0, 0, 0, 0, 0, 6, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
             [0, 0, 2, 8, 2, 3, 7, 0, 0, 10], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
             [0, 0, 0, 4, 0, 2, 9, 4, 0, 5], [6, 3, 0, 8, 0, 10, 0, 9, 4, 0]])

        tensor = Tensor.fromRandom(None, shape, [0.5, 0.5], 10, seed=3)

        self.assertEqual(tensor, tensor_ref)
        self.assertEqual(tensor.getRankIds(), rank_ids)