tensor = torch.rand((8, 3, 384, 640)) # 生成一个形状为(5,5,5,5)的随机Tensor作为示例
# 获取tensor的大小
n1, n2, n3, n4 = tensor.size()
# 产生10个随机索引
rand_indices = torch.randint(0, min(n1 * n2 * n3 * n4, 10), (10,))
print(rand_indices)
# 根据随机索引获取元素
rand_elements = tensor.view(-1)[rand_indices]
print(rand_elements)
Python gets torch tensor random 10 positions
Guess you like
Origin blog.csdn.net/jacke121/article/details/131553205
Recommended
Ranking