小编Atu*_*yak的帖子

例如，我需要创建一个像这样的 3D 张量 (5,3,2)

array([[[0, 0],
        [0, 1],
        [0, 0]],

       [[1, 0],
        [0, 0],
        [0, 0]],

       [[0, 0],
        [1, 0],
        [0, 0]],

       [[0, 0],
        [0, 0],
        [1, 0]],

       [[0, 0],
        [0, 1],
        [0, 0]]])

每个切片中都应该随机放置一个“1”（如果您将张量视为一条面包）。这可以使用循环来完成，但我想向量化这部分。