bethgelab / foolbox / 8137716344

coverage: 98.47%. Remained the same

Source File
Press 'n' to go to next uncovered line, 'b' for previous

100.0
/foolbox/devutils.py
1 
"""Internal module with utility functions"""
2 
import eagerpy as ep
10✔
3 









4












5



def flatten(x: ep.Tensor, keep: int = 1) -> ep.Tensor:




10✔





6



    return x.flatten(start=keep)




8✔





7












8












9



def atleast_kd(x: ep.Tensor, k: int) -> ep.Tensor:




10✔





10



    shape = x.shape + (1,) * (k - x.ndim)




8✔





11



    return x.reshape(shape)




8✔



















    

  • 

    •