paddingMask static method

Tensor paddingMask(
  1. List<int> lengths,
  2. int maxLen
)

Create a padding mask (1 for valid, 0 for padding).

Implementation

static Tensor paddingMask(List<int> lengths, int maxLen) {
  final batchSize = lengths.length;
  final data = Float32List(batchSize * maxLen);
  for (int b = 0; b < batchSize; b++) {
    for (int i = 0; i < maxLen; i++) {
      data[b * maxLen + i] = i < lengths[b] ? 1.0 : 0.0;
    }
  }
  return Tensor(data, [batchSize, maxLen]);
}