tensor_utils

Functions

add_dummy_dims_left(x, y)

add_dummy_dims_right(x, y)

broadcast(x, y, ignored_dims)

Parameters:

ignored_dims (List[int])

broadcast_all(xs, ignored_dims=[])

broadcast_both(x, y, ignored_dims)

Parameters:

ignored_dims (List[int])

broadcast_both_left(x, y, ignored_dims)

Parameters:

ignored_dims (List[int])

broadcast_cross_product(x, y, dim=-1)

broadcast_gather(src, dim, ind, keepdim=False, **kwargs)

Parameters:

dim (int)

broadcast_interleave(x, counts, inds, dim=-2)

broadcast_scatter(input, dim, ind, src, **kwargs)

cast_to_tensor(x)

Converts scalars or lists of scalars into tensors, and combines lists of tensors into a single tensor. All other input types are returned unchanged. Returned tensors are always of shape [1,N,D] where D is dimension and N is the number of tensors combined.

cast_to_tensor_single(x)

concat_dicts(kwargs)

Concatenates a list of dicts sharing the same keys, the resulting dictionary has the same keys and concatenated values.

dot_product(x, y, dim=-1, keepdim=True, out=None)

dot_product_in_place(x, y, dim=-1)

expand_as_left(x, y, offset=0)

Parameters:

offset (int)

expand_as_right(x, y, offset=0)

Parameters:

offset (int)

implements(torch_function)

Register a torch function override for ScalarTensor

interpolate(x, y, a)

make_grid(height, width=None, min_coord=-1, max_coord=1, min_coord2=None, max_coord2=None)

mean(xs)

mid_point(x, dim=-1, keepdim=True)

offset(x)

pack_tensor(x, packing)

packed_reorder(x, counts, ids)

pad_dim_left(x, num_dims)

pad_dim_right(x, num_dims)

pad_dims(xs, unsqueeze_dim=-2)

prepare_kwargs(self, func, args, kwargs, initial_args, unique_args)

Combine args and kwargs into one dict, using default values where arg is missing

reduce_max_score(x, scores, dim=-1)

robust_concat(xs)

Concatenates multiple tensors together while broadcasting as necessary to ensure shapes match.

scatter_arg_max(x, inds, dim=-1, dim_size=None)

shuffle(x)

squish(x, start=0, end=1)

Parameters:

• start (int)

• end (int)

unpack_tensor(x, packing)

unsqueeze_dims(x, y, insert_dim=0)

unsqueeze_left(x, y, offset=0)

Parameters:

offset (int)

unsqueeze_pack_tensors(xs, packing)

unsqueeze_right(x, y, offset=0)

Parameters:

offset (int)

unsqueeze_until_dim(x, dim, insert_dim=0)

unsquish(x, dim=0, factor=None)

Parameters:

• dim (int)

• factor (int | None)