Python resize2D_asの例

コード例 #1

ファイル: adf.py プロジェクト: SaumilShah66/Drone_RL

def concatenate_as(tensor_list, tensor_as, dim, mode="bilinear"):
    means = [resize2D_as(x[0], tensor_as[0], mode=mode) for x in tensor_list]
    variances = [
        resize2D_as(x[1], tensor_as[0], mode=mode) for x in tensor_list
    ]
    means = torch.cat(means, dim=dim)
    variances = torch.cat(variances, dim=dim)
    return means, variances

コード例 #2

def concatenate_as(tensor_list, tensor_as, dim, mode="bilinear"):
    tensor_list = [resize2D_as(x, tensor_as, mode=mode) for x in tensor_list]
    return torch.cat(tensor_list, dim=dim)