tf.gather与torch.gather的区别与认识

tf.gather(
    params,                  #输入值
    indices,                 #索引
    validate_indices=None, 
    name=None, 
    axis=None,               #用来收集索引的值
    batch_dims=0
)

torch.gather(input,            #输入值
             dim,              #索引的轴
             index,            #索引
             out=None, 
             sparse_grad=False)