tensorflow: Allow tf.Tensor to be used in __getitem__ (#13549)

This commit is contained in:
Hoël Bagard
2025-02-27 20:19:20 +09:00
committed by GitHub
parent 62feb28c29
commit 8c6493925f
+1 -1
View File
@@ -32,7 +32,7 @@ KerasSerializable: TypeAlias = KerasSerializable1 | KerasSerializable2
TensorValue: TypeAlias = tf.Tensor # Alias for a 0D Tensor
Integer: TypeAlias = TensorValue | int | IntArray | np.number[Any] # Here IntArray are assumed to be 0D.
Float: TypeAlias = Integer | float | FloatArray
Slice: TypeAlias = int | slice | None
Slice: TypeAlias = tf.Tensor | tf.RaggedTensor | int | slice | None
FloatDataSequence: TypeAlias = Sequence[float] | Sequence[FloatDataSequence]
IntDataSequence: TypeAlias = Sequence[int] | Sequence[IntDataSequence]
StrDataSequence: TypeAlias = Sequence[str] | Sequence[StrDataSequence]