allennlp.nn.util¶
Assorted utilities for working with neural networks in AllenNLP.

allennlp.nn.util.
add_sentence_boundary_token_ids
(tensor: torch.FloatTensor, mask: torch.FloatTensor, sentence_begin_token: typing.Any, sentence_end_token: typing.Any) → typing.Tuple[torch.FloatTensor, torch.FloatTensor][source]¶ Add begin/end of sentence tokens to the batch of sentences. Given a batch of sentences with size
(batch_size, timesteps)
or(batch_size, timesteps, dim)
this returns a tensor of shape(batch_size, timesteps + 2)
or(batch_size, timesteps + 2, dim)
respectively.Returns both the new tensor and updated mask.
Parameters: tensor :
torch.Tensor
A tensor of shape
(batch_size, timesteps)
or(batch_size, timesteps, dim)
mask :
torch.Tensor
A tensor of shape
(batch_size, timesteps)
sentence_begin_token: Any (anything that can be broadcast in torch for assignment)
For 2D input, a scalar with the <S> id. For 3D input, a tensor with length dim.
sentence_end_token: Any (anything that can be broadcast in torch for assignment)
For 2D input, a scalar with the </S> id. For 3D input, a tensor with length dim.
Returns: tensor_with_boundary_tokens :
torch.Tensor
The tensor with the appended and prepended boundary tokens. If the input was 2D, it has shape (batch_size, timesteps + 2) and if the input was 3D, it has shape (batch_size, timesteps + 2, dim).
new_mask :
torch.Tensor
The new mask for the tensor, taking into account the appended tokens marking the beginning and end of the sentence.

allennlp.nn.util.
batched_index_select
(target: torch.FloatTensor, indices: torch.LongTensor, flattened_indices: typing.Union[torch.LongTensor, NoneType] = None) → torch.FloatTensor[source]¶ The given indices of size
(batch_size, d_1, ..., d_n)
indexes into the sequence dimension (dimension 2) of the target, which has size(batch_size, sequence_length, embedding_size)
.This function returns selected values in the target with respect to the provided indices, which have size
(batch_size, d_1, ..., d_n, embedding_size)
. This can use the optionally precomputedflattened_indices()
with size(batch_size * d_1 * ... * d_n)
if given.An example use case of this function is looking up the start and end indices of spans in a sequence tensor. This is used in the
CoreferenceResolver
. Model to select contextual word representations corresponding to the start and end indices of mentions. The key reason this can’t be done with basic torch functions is that we want to be able to use lookup tensors with an arbitrary number of dimensions (for example, in the coref model, we don’t know apriori how many spans we are looking up).Parameters: target :
torch.Tensor
, required.A 3 dimensional tensor of shape (batch_size, sequence_length, embedding_size). This is the tensor to be indexed.
indices :
torch.LongTensor
A tensor of shape (batch_size, ...), where each element is an index into the
sequence_length
dimension of thetarget
tensor.flattened_indices : Optional[torch.Tensor], optional (default = None)
An optional tensor representing the result of calling :func:~`flatten_and_batch_shift_indices` on
indices
. This is helpful in the case that the indices can be flattened once and cached for many batch lookups.Returns: selected_targets :
torch.Tensor
A tensor with shape [indices.size(), target.size(1)] representing the embedded indices extracted from the batch flattened target tensor.

allennlp.nn.util.
bucket_values
(distances: torch.FloatTensor, num_identity_buckets: int = 4, num_total_buckets: int = 10) → torch.FloatTensor[source]¶ Places the given values (designed for distances) into
num_total_buckets``semilogscale buckets, with ``num_identity_buckets
of these capturing single values.The default settings will bucket values into the following buckets: [0, 1, 2, 3, 4, 57, 815, 1631, 3263, 64+].
Parameters: distances :
torch.Tensor
, required.A Tensor of any size, to be bucketed.
num_identity_buckets: int, optional (default = 4).
The number of identity buckets (those only holding a single value).
num_total_buckets : int, (default = 10)
The total number of buckets to bucket values into.
Returns: A tensor of the same shape as the input, containing the indices of the buckets
the values were placed in.

allennlp.nn.util.
combine_tensors
(combination: str, tensors: typing.List[torch.FloatTensor]) → torch.FloatTensor[source]¶ Combines a list of tensors using elementwise operations and concatenation, specified by a
combination
string. The string refers to (1indexed) positions in the input tensor list, and looks like"1,2,1+2,31"
.We allow the following kinds of combinations:
x
,x*y
,x+y
,xy
, andx/y
, wherex
andy
are positive integers less than or equal tolen(tensors)
. Each of the binary operations is performed elementwise. You can give as many combinations as you want in thecombination
string. For example, for the input string"1,2,1*2"
, the result would be[1;2;1*2]
, as you would expect, where[;]
is concatenation along the last dimension.If you have a fixed, known way to combine tensors that you use in a model, you should probably just use something like
torch.cat([x_tensor, y_tensor, x_tensor * y_tensor])
. This function adds some complexity that is only necessary if you want the specific combination used to be configurable.If you want to do any elementwise operations, the tensors involved in each elementwise operation must have the same shape.
This function also accepts
x
andy
in place of1
and2
in the combination string.

allennlp.nn.util.
device_mapping
(cuda_device: int)[source]¶ In order to torch.load() a GPUtrained model onto a CPU (or specific GPU), you have to supply a map_location function. Call this with the desired cuda_device to get the function that torch.load() needs.

allennlp.nn.util.
flatten_and_batch_shift_indices
(indices: torch.FloatTensor, sequence_length: int) → torch.FloatTensor[source]¶ This is a subroutine for
batched_index_select()
. The givenindices
of size(batch_size, d_1, ..., d_n)
indexes into dimension 2 of a target tensor, which has size(batch_size, sequence_length, embedding_size)
. This function returns a vector that correctly indexes into the flattened target. The sequence length of the target must be provided to compute the appropriate offsets.indices = torch.ones([2,3]).long() # Sequence length of the target tensor. sequence_length = 10 shifted_indices = flatten_and_batch_shift_indices(indices, sequence_length) # Indices into the second element in the batch are correctly shifted # to take into account that the target tensor will be flattened before # the indices are applied. assert shifted_indices == [1, 1, 1, 11, 11, 11]
Parameters: indices :
torch.LongTensor
, required.sequence_length :
int
, required.The length of the sequence the indices index into. This must be the second dimension of the tensor.
Returns: offset_indices :
torch.LongTensor

allennlp.nn.util.
flattened_index_select
(target: torch.FloatTensor, indices: torch.LongTensor) → torch.FloatTensor[source]¶ The given
indices
of size(set_size, subset_size)
specifies subsets of thetarget
that each of the set_size rows should select. The target has size(batch_size, sequence_length, embedding_size)
, and the resulting selected tensor has size(batch_size, set_size, subset_size, embedding_size)
.Parameters: target :
torch.Tensor
, required.A Tensor of shape (batch_size, sequence_length, embedding_size).
indices :
torch.LongTensor
, required.A LongTensor of shape (set_size, subset_size). All indices must be < sequence_length as this tensor is an index into the sequence_length dimension of the target.
Returns: selected :
torch.Tensor
, required.A Tensor of shape (batch_size, set_size, subset_size, embedding_size).

allennlp.nn.util.
get_combined_dim
(combination: str, tensor_dims: typing.List[int]) → int[source]¶ For use with
combine_tensors()
. This function computes the resultant dimension when callingcombine_tensors(combination, tensors)
, when the tensor dimension is known. This is necessary for knowing the sizes of weight matrices when building models that usecombine_tensors
.Parameters: combination :
str
A commaseparated list of combination pieces, like
"1,2,1*2"
, specified identically tocombination
incombine_tensors()
.tensor_dims :
List[int]
A list of tensor dimensions, where each dimension is from the last axis of the tensors that will be input to
combine_tensors()
.

allennlp.nn.util.
get_dropout_mask
(dropout_probability: float, tensor_for_masking: torch.autograd.variable.Variable)[source]¶ Computes and returns an elementwise dropout mask for a given tensor, where each element in the mask is dropped out with probability dropout_probability. Note that the mask is NOT applied to the tensor  the tensor is passed to retain the correct CUDA tensor type for the mask.
Parameters: dropout_probability : float, required.
Probability of dropping a dimension of the input.
tensor_for_masking : torch.Variable, required.
Returns: A torch.FloatTensor consisting of the binary mask scaled by 1/ (1  dropout_probability).
This scaling ensures expected values and variances of the output of applying this mask
and the original tensor are the same.

allennlp.nn.util.
get_lengths_from_binary_sequence_mask
(mask: torch.FloatTensor)[source]¶ Compute sequence lengths for each batch element in a tensor using a binary mask.
Parameters: mask : torch.Tensor, required.
A 2D binary mask of shape (batch_size, sequence_length) to calculate the perbatch sequence lengths from.
Returns: A torch.LongTensor of shape (batch_size,) representing the lengths
of the sequences in the batch.

allennlp.nn.util.
get_range_vector
(size: int, is_cuda: bool) → torch.FloatTensor[source]¶ Returns a range vector with the desired size, starting at 0. The CUDA implementation is meant to avoid copy data from CPU to GPU.

allennlp.nn.util.
get_text_field_mask
(text_field_tensors: typing.Dict[str, torch.FloatTensor], num_wrapping_dims: int = 0) → torch.LongTensor[source]¶ Takes the dictionary of tensors produced by a
TextField
and returns a mask with 0 where the tokens are padding, and 1 otherwise. We also handleTextFields
wrapped by an arbitrary number ofListFields
, where the number of wrappingListFields
is given bynum_wrapping_dims
.If
num_wrapping_dims == 0
, the returned mask has shape(batch_size, num_tokens)
. Ifnum_wrapping_dims > 0
then the returned mask hasnum_wrapping_dims
extra dimensions, so the shape will be(batch_size, ..., num_tokens)
.There could be several entries in the tensor dictionary with different shapes (e.g., one for word ids, one for character ids). In order to get a token mask, we use the tensor in the dictionary with the lowest number of dimensions. After subtracting
num_wrapping_dims
, if this tensor has two dimensions we assume it has shape(batch_size, ..., num_tokens)
, and use it for the mask. If instead it has three dimensions, we assume it has shape(batch_size, ..., num_tokens, num_features)
, and sum over the last dimension to produce the mask. Most frequently this will be a character id tensor, but it could also be a featurized representation of each token, etc.NOTE: Our functions for generating masks create torch.LongTensors, because using torch.ByteTensors inside Variables makes it easy to run into overflow errors when doing mask manipulation, such as summing to get the lengths of sequences  see below. >>> mask = torch.ones([260]).byte() >>> mask.sum() # equals 260. >>> var_mask = torch.autograd.Variable(mask) >>> var_mask.sum() # equals 4, due to 8 bit precision  the sum overflows.

allennlp.nn.util.
last_dim_log_softmax
(tensor: torch.FloatTensor, mask: typing.Union[torch.FloatTensor, NoneType] = None) → torch.FloatTensor[source]¶ Takes a tensor with 3 or more dimensions and does a masked log softmax over the last dimension. We assume the tensor has shape
(batch_size, ..., sequence_length)
and that the mask (if given) has shape(batch_size, sequence_length)
.

allennlp.nn.util.
last_dim_softmax
(tensor: torch.FloatTensor, mask: typing.Union[torch.FloatTensor, NoneType] = None) → torch.FloatTensor[source]¶ Takes a tensor with 3 or more dimensions and does a masked softmax over the last dimension. We assume the tensor has shape
(batch_size, ..., sequence_length)
and that the mask (if given) has shape(batch_size, sequence_length)
.

allennlp.nn.util.
logsumexp
(tensor: torch.FloatTensor, dim: int = 1, keepdim: bool = False) → torch.FloatTensor[source]¶ A numerically stable computation of logsumexp. This is mathematically equivalent to tensor.exp().sum(dim, keep=keepdim).log(). This function is typically used for summing log probabilities.
Parameters: tensor : torch.FloatTensor, required.
A tensor of arbitrary size.
dim : int, optional (default = 1)
The dimension of the tensor to apply the logsumexp to.
keepdim: bool, optional (default = False)
Whether to retain a dimension of size one at the dimension we reduce over.

allennlp.nn.util.
masked_log_softmax
(vector, mask)[source]¶ torch.nn.functional.log_softmax(vector)
does not work if some elements ofvector
should be masked. This performs a log_softmax on just the nonmasked portions ofvector
. PassingNone
in for the mask is also acceptable; you’ll just get a regular log_softmax.We assume that both
vector
andmask
(if given) have shape(batch_size, vector_dim)
.In the case that the input vector is completely masked, this function returns an array of
0.0
. You should be masking the result of whatever computation comes out of this in that case, anyway, so it shouldn’t matter.

allennlp.nn.util.
masked_softmax
(vector, mask)[source]¶ torch.nn.functional.softmax(vector)
does not work if some elements ofvector
should be masked. This performs a softmax on just the nonmasked portions ofvector
. PassingNone
in for the mask is also acceptable; you’ll just get a regular softmax.We assume that both
vector
andmask
(if given) have shape(batch_size, vector_dim)
.In the case that the input vector is completely masked, this function returns an array of
0.0
. This behavior may causeNaN
if this is used as the last layer of a model that uses categorical crossentropy loss.

allennlp.nn.util.
ones_like
(tensor: torch.FloatTensor) → torch.FloatTensor[source]¶ Use clone() + fill_() to make sure that a ones tensor ends up on the right device at runtime.

allennlp.nn.util.
remove_sentence_boundaries
(tensor: torch.FloatTensor, mask: torch.FloatTensor) → typing.Tuple[torch.FloatTensor, torch.FloatTensor][source]¶ Remove begin/end of sentence embeddings from the batch of sentences. Given a batch of sentences with size
(batch_size, timesteps, dim)
this returns a tensor of shape(batch_size, timesteps  2, dim)
after removing the beginning and end sentence markers. The sentences are assumed to be padded on the right, with the beginning of each sentence assumed to occur at index 0 (i.e.,mask[:, 0]
is assumed to be 1).Returns both the new tensor and updated mask.
This function is the inverse of
add_sentence_boundary_token_ids
.Parameters: tensor :
torch.Tensor
A tensor of shape
(batch_size, timesteps, dim)
mask :
torch.Tensor
A tensor of shape
(batch_size, timesteps)
Returns: tensor_without_boundary_tokens :
torch.Tensor
The tensor after removing the boundary tokens of shape
(batch_size, timesteps  2, dim)
new_mask :
torch.Tensor
The new mask for the tensor of shape
(batch_size, timesteps  2)
.

allennlp.nn.util.
replace_masked_values
(tensor: torch.autograd.variable.Variable, mask: torch.autograd.variable.Variable, replace_with: float) → torch.autograd.variable.Variable[source]¶ Replaces all masked values in
tensor
withreplace_with
.mask
must be broadcastable to the same shape astensor
. We require thattensor.dim() == mask.dim()
, as otherwise we won’t know which dimensions of the mask to unsqueeze.

allennlp.nn.util.
sequence_cross_entropy_with_logits
(logits: torch.FloatTensor, targets: torch.LongTensor, weights: torch.FloatTensor, batch_average: bool = True) → torch.FloatTensor[source]¶ Computes the cross entropy loss of a sequence, weighted with respect to some user provided weights. Note that the weighting here is not the same as in the
torch.nn.CrossEntropyLoss()
criterion, which is weighting classes; here we are weighting the loss contribution from particular elements in the sequence. This allows loss computations for models which use padding.Parameters: logits :
torch.FloatTensor
, required.A
torch.FloatTensor
of size (batch_size, sequence_length, num_classes) which contains the unnormalized probability for each class.targets :
torch.LongTensor
, required.A
torch.LongTensor
of size (batch, sequence_length) which contains the index of the true class for each corresponding step.weights :
torch.FloatTensor
, required.A
torch.FloatTensor
of size (batch, sequence_length)batch_average : bool, optional, (default = True).
A bool indicating whether the loss should be averaged across the batch, or returned as a vector of losses per batch element.
Returns: A torch.FloatTensor representing the cross entropy loss.
If
batch_average == True
, the returned loss is a scalar.If
batch_average == False
, the returned loss is a vector of shape (batch_size,).

allennlp.nn.util.
sort_batch_by_length
(tensor: torch.autograd.variable.Variable, sequence_lengths: torch.autograd.variable.Variable)[source]¶ Sort a batch first tensor by some specified lengths.
Parameters: tensor : Variable(torch.FloatTensor), required.
A batch first Pytorch tensor.
sequence_lengths : Variable(torch.LongTensor), required.
A tensor representing the lengths of some dimension of the tensor which we want to sort by.
Returns: sorted_tensor : Variable(torch.FloatTensor)
The original tensor sorted along the batch dimension with respect to sequence_lengths.
sorted_sequence_lengths : Variable(torch.LongTensor)
The original sequence_lengths sorted by decreasing size.
restoration_indices : Variable(torch.LongTensor)
Indices into the sorted_tensor such that
sorted_tensor.index_select(0, restoration_indices) == original_tensor
permuation_index : Variable(torch.LongTensor)
The indices used to sort the tensor. This is useful if you want to sort many tensors using the same ordering.

allennlp.nn.util.
viterbi_decode
(tag_sequence: torch.FloatTensor, transition_matrix: torch.FloatTensor, tag_observations: typing.Union[typing.List[int], NoneType] = None)[source]¶ Perform Viterbi decoding in log space over a sequence given a transition matrix specifying pairwise (transition) potentials between tags and a matrix of shape (sequence_length, num_tags) specifying unary potentials for possible tags per timestep.
Parameters: tag_sequence : torch.Tensor, required.
A tensor of shape (sequence_length, num_tags) representing scores for a set of tags over a given sequence.
transition_matrix : torch.Tensor, required.
A tensor of shape (num_tags, num_tags) representing the binary potentials for transitioning between a given pair of tags.
tag_observations : Optional[List[int]], optional, (default = None)
A list of length
sequence_length
containing the class ids of observed elements in the sequence, with unobserved elements being set to 1. Note that it is possible to provide evidence which results in degenerate labellings if the sequences of tags you provide as evidence cannot transition between each other, or those transitions are extremely unlikely. In this situation we log a warning, but the responsibility for providing selfconsistent evidence ultimately lies with the user.Returns: viterbi_path : List[int]
The tag indices of the maximum likelihood tag sequence.
viterbi_score : float
The score of the viterbi path.

allennlp.nn.util.
weighted_sum
(matrix: torch.FloatTensor, attention: torch.FloatTensor) → torch.FloatTensor[source]¶ Takes a matrix of vectors and a set of weights over the rows in the matrix (which we call an “attention” vector), and returns a weighted sum of the rows in the matrix. This is the typical computation performed after an attention mechanism.
Note that while we call this a “matrix” of vectors and an attention “vector”, we also handle higherorder tensors. We always sum over the secondtolast dimension of the “matrix”, and we assume that all dimensions in the “matrix” prior to the last dimension are matched in the “vector”. Nonmatched dimensions in the “vector” must be directly after the batch dimension.
For example, say I have a “matrix” with dimensions
(batch_size, num_queries, num_words, embedding_dim)
. The attention “vector” then must have at least those dimensions, and could have more. Both:(batch_size, num_queries, num_words)
(distribution over words for each query)(batch_size, num_documents, num_queries, num_words)
(distribution over words in a query for each document)
are valid input “vectors”, producing tensors of shape:
(batch_size, num_queries, embedding_dim)
and(batch_size, num_documents, num_queries, embedding_dim)
respectively.