def feats_at_batch(stensor, batch_id):
    """get features of sparse tensor at batch id
    Args:
        - tensor (Sparse Tensor)
        - batch_id (int)
    Returns:
        - (Tensor)
    """
    mask = stensor.indices[:, 0] == batch_id
    return stensor.features[mask]