torch.le(input, other, out=None) → Tensor 逐元素比较input和other , 即是否input<=other
返回1或0的tensor结果
index_select(x, 1, indices,out) →Tensor
1代表维度1,即列,indices是筛选的索引序号。x输入,out输出对应x的索引
torch.clamp(input, min, max, out=None) → Tensor
将输入input
张量每个元素的夹紧到区间 [min,max],并返回结果到一个新张量。