deepfold.utils.tensor_utils

Utilities related to tensor operations.

Functions

add(m1, m2, inplace)
batched_gather(data, inds[, dim, num_batch_dims])
collate(samples)	Converts list of samples into a batch dict.
dict_map(fn, dic, leaf_type)
dict_multimap(fn, dicts)
flatten_final_dims(t, num_dims)
masked_mean(mask, value, dim[, eps, keepdim])
one_hot(x, v_bins)
permute_final_dims(tensor, inds)
pts_to_distogram(pts[, min_bin, max_bin, ...])
tree_map()