创建一个三维零张量，在numpy/jax中的每个切片上随机放置一个'1‘

import numpy as np

# 使用numpy创建三维零张量
tensor = np.zeros((3, 3, 3))

import numpy as np

# 使用numpy创建三维零张量
tensor = np.zeros((3, 3, 3))

# 在每个切片上随机放置一个'1'
for i in range(tensor.shape[0]):
    slice = tensor[i]
    random_index = np.random.randint(0, slice.size)
    slice.flat[random_index] = 1