python 三维list选取特殊条件的值输出

比如，此时条件为：

三维torch tensor　a　中输出每个二维tensor中每行最后一个数大于 0.5 时的所有内容，输出形式不能变，使用list输出
显然，此时输出的 list 应为三维形式，与 torch.tensor 的区别在于其每个二维列表的0维无法保证相等。

a = a.cpu().numpy()
aaa = []
for i in range(0,6):
    aa = []
    for j,data in enumerate(a[i,:,-1]):
        if data > 0.5:
            b = a.tolist()
            aa.append(b[i][j])
        else: continue
    aaa.append(aa)
print(aaa)