foscat.loss_backend_tens#

Classes#

Module Contents#

class foscat.loss_backend_tens.loss_backend(backend, curr_gpu, mpi_rank)#

bk#

curr_gpu#

mpi_rank#

check_dense(data, datasz)#

loss(x, batch, loss_function, KEEP_TRACK)#