(features, attribute, dtype, device)
| 116 | return filtered_sentence_tag |
| 117 | |
| 118 | def create_tensor(features, attribute, dtype, device): |
| 119 | try: |
| 120 | return torch.tensor([getattr(f, attribute) for f in features], dtype=dtype).to(device) |
| 121 | except Exception as e: |
| 122 | if attribute not in create_tensor.attribute_warning: |
| 123 | print("Exception in attribute {}".format(attribute)) |
| 124 | print(e) |
| 125 | create_tensor.attribute_warning.add(attribute) |
| 126 | return None |
| 127 | |
| 128 | create_tensor.attribute_warning = set([]) |
| 129 |