zea 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
zea/keras_ops.py ADDED
@@ -0,0 +1,1989 @@
1
+ """Auto-generated :class:`zea.Operation` for all unary :mod:`keras.ops`
2
+ and :mod:`keras.ops.image` functions.
3
+
4
+ They can be used in zea pipelines like any other :class:`zea.Operation`, for example:
5
+
6
+ .. code-block:: python
7
+
8
+ from zea.keras_ops import Squeeze
9
+
10
+ op = Squeeze(axis=1)
11
+
12
+ This file is generated automatically. Do not edit manually.
13
+ Generated with Keras 3.11.3
14
+ """
15
+
16
+ import keras
17
+
18
+ from zea.internal.registry import ops_registry
19
+ from zea.ops import Lambda
20
+
21
+ class MissingKerasOps(ValueError):
22
+ def __init__(self, class_name: str, func: str):
23
+ super().__init__(
24
+ f"Failed to create {class_name} with {func}. " +
25
+ "This may be due to an incompatible version of `keras`. " +
26
+ "Please try to upgrade `keras` to the latest version by running " +
27
+ "`pip install --upgrade keras`."
28
+ )
29
+
30
+
31
+ @ops_registry("keras.ops.abs")
32
+ class Abs(Lambda):
33
+ """Operation wrapping keras.ops.abs."""
34
+
35
+ def __init__(self, **kwargs):
36
+ try:
37
+ super().__init__(func=keras.ops.abs, **kwargs)
38
+ except AttributeError as e:
39
+ raise MissingKerasOps("Abs", "keras.ops.abs") from e
40
+
41
+ @ops_registry("keras.ops.absolute")
42
+ class Absolute(Lambda):
43
+ """Operation wrapping keras.ops.absolute."""
44
+
45
+ def __init__(self, **kwargs):
46
+ try:
47
+ super().__init__(func=keras.ops.absolute, **kwargs)
48
+ except AttributeError as e:
49
+ raise MissingKerasOps("Absolute", "keras.ops.absolute") from e
50
+
51
+ @ops_registry("keras.ops.all")
52
+ class All(Lambda):
53
+ """Operation wrapping keras.ops.all."""
54
+
55
+ def __init__(self, **kwargs):
56
+ try:
57
+ super().__init__(func=keras.ops.all, **kwargs)
58
+ except AttributeError as e:
59
+ raise MissingKerasOps("All", "keras.ops.all") from e
60
+
61
+ @ops_registry("keras.ops.amax")
62
+ class Amax(Lambda):
63
+ """Operation wrapping keras.ops.amax."""
64
+
65
+ def __init__(self, **kwargs):
66
+ try:
67
+ super().__init__(func=keras.ops.amax, **kwargs)
68
+ except AttributeError as e:
69
+ raise MissingKerasOps("Amax", "keras.ops.amax") from e
70
+
71
+ @ops_registry("keras.ops.amin")
72
+ class Amin(Lambda):
73
+ """Operation wrapping keras.ops.amin."""
74
+
75
+ def __init__(self, **kwargs):
76
+ try:
77
+ super().__init__(func=keras.ops.amin, **kwargs)
78
+ except AttributeError as e:
79
+ raise MissingKerasOps("Amin", "keras.ops.amin") from e
80
+
81
+ @ops_registry("keras.ops.angle")
82
+ class Angle(Lambda):
83
+ """Operation wrapping keras.ops.angle."""
84
+
85
+ def __init__(self, **kwargs):
86
+ try:
87
+ super().__init__(func=keras.ops.angle, **kwargs)
88
+ except AttributeError as e:
89
+ raise MissingKerasOps("Angle", "keras.ops.angle") from e
90
+
91
+ @ops_registry("keras.ops.any")
92
+ class Any(Lambda):
93
+ """Operation wrapping keras.ops.any."""
94
+
95
+ def __init__(self, **kwargs):
96
+ try:
97
+ super().__init__(func=keras.ops.any, **kwargs)
98
+ except AttributeError as e:
99
+ raise MissingKerasOps("Any", "keras.ops.any") from e
100
+
101
+ @ops_registry("keras.ops.arccos")
102
+ class Arccos(Lambda):
103
+ """Operation wrapping keras.ops.arccos."""
104
+
105
+ def __init__(self, **kwargs):
106
+ try:
107
+ super().__init__(func=keras.ops.arccos, **kwargs)
108
+ except AttributeError as e:
109
+ raise MissingKerasOps("Arccos", "keras.ops.arccos") from e
110
+
111
+ @ops_registry("keras.ops.arccosh")
112
+ class Arccosh(Lambda):
113
+ """Operation wrapping keras.ops.arccosh."""
114
+
115
+ def __init__(self, **kwargs):
116
+ try:
117
+ super().__init__(func=keras.ops.arccosh, **kwargs)
118
+ except AttributeError as e:
119
+ raise MissingKerasOps("Arccosh", "keras.ops.arccosh") from e
120
+
121
+ @ops_registry("keras.ops.arcsin")
122
+ class Arcsin(Lambda):
123
+ """Operation wrapping keras.ops.arcsin."""
124
+
125
+ def __init__(self, **kwargs):
126
+ try:
127
+ super().__init__(func=keras.ops.arcsin, **kwargs)
128
+ except AttributeError as e:
129
+ raise MissingKerasOps("Arcsin", "keras.ops.arcsin") from e
130
+
131
+ @ops_registry("keras.ops.arcsinh")
132
+ class Arcsinh(Lambda):
133
+ """Operation wrapping keras.ops.arcsinh."""
134
+
135
+ def __init__(self, **kwargs):
136
+ try:
137
+ super().__init__(func=keras.ops.arcsinh, **kwargs)
138
+ except AttributeError as e:
139
+ raise MissingKerasOps("Arcsinh", "keras.ops.arcsinh") from e
140
+
141
+ @ops_registry("keras.ops.arctan")
142
+ class Arctan(Lambda):
143
+ """Operation wrapping keras.ops.arctan."""
144
+
145
+ def __init__(self, **kwargs):
146
+ try:
147
+ super().__init__(func=keras.ops.arctan, **kwargs)
148
+ except AttributeError as e:
149
+ raise MissingKerasOps("Arctan", "keras.ops.arctan") from e
150
+
151
+ @ops_registry("keras.ops.arctanh")
152
+ class Arctanh(Lambda):
153
+ """Operation wrapping keras.ops.arctanh."""
154
+
155
+ def __init__(self, **kwargs):
156
+ try:
157
+ super().__init__(func=keras.ops.arctanh, **kwargs)
158
+ except AttributeError as e:
159
+ raise MissingKerasOps("Arctanh", "keras.ops.arctanh") from e
160
+
161
+ @ops_registry("keras.ops.argmax")
162
+ class Argmax(Lambda):
163
+ """Operation wrapping keras.ops.argmax."""
164
+
165
+ def __init__(self, **kwargs):
166
+ try:
167
+ super().__init__(func=keras.ops.argmax, **kwargs)
168
+ except AttributeError as e:
169
+ raise MissingKerasOps("Argmax", "keras.ops.argmax") from e
170
+
171
+ @ops_registry("keras.ops.argmin")
172
+ class Argmin(Lambda):
173
+ """Operation wrapping keras.ops.argmin."""
174
+
175
+ def __init__(self, **kwargs):
176
+ try:
177
+ super().__init__(func=keras.ops.argmin, **kwargs)
178
+ except AttributeError as e:
179
+ raise MissingKerasOps("Argmin", "keras.ops.argmin") from e
180
+
181
+ @ops_registry("keras.ops.argpartition")
182
+ class Argpartition(Lambda):
183
+ """Operation wrapping keras.ops.argpartition."""
184
+
185
+ def __init__(self, **kwargs):
186
+ try:
187
+ super().__init__(func=keras.ops.argpartition, **kwargs)
188
+ except AttributeError as e:
189
+ raise MissingKerasOps("Argpartition", "keras.ops.argpartition") from e
190
+
191
+ @ops_registry("keras.ops.argsort")
192
+ class Argsort(Lambda):
193
+ """Operation wrapping keras.ops.argsort."""
194
+
195
+ def __init__(self, **kwargs):
196
+ try:
197
+ super().__init__(func=keras.ops.argsort, **kwargs)
198
+ except AttributeError as e:
199
+ raise MissingKerasOps("Argsort", "keras.ops.argsort") from e
200
+
201
+ @ops_registry("keras.ops.array")
202
+ class Array(Lambda):
203
+ """Operation wrapping keras.ops.array."""
204
+
205
+ def __init__(self, **kwargs):
206
+ try:
207
+ super().__init__(func=keras.ops.array, **kwargs)
208
+ except AttributeError as e:
209
+ raise MissingKerasOps("Array", "keras.ops.array") from e
210
+
211
+ @ops_registry("keras.ops.average")
212
+ class Average(Lambda):
213
+ """Operation wrapping keras.ops.average."""
214
+
215
+ def __init__(self, **kwargs):
216
+ try:
217
+ super().__init__(func=keras.ops.average, **kwargs)
218
+ except AttributeError as e:
219
+ raise MissingKerasOps("Average", "keras.ops.average") from e
220
+
221
+ @ops_registry("keras.ops.bartlett")
222
+ class Bartlett(Lambda):
223
+ """Operation wrapping keras.ops.bartlett."""
224
+
225
+ def __init__(self, **kwargs):
226
+ try:
227
+ super().__init__(func=keras.ops.bartlett, **kwargs)
228
+ except AttributeError as e:
229
+ raise MissingKerasOps("Bartlett", "keras.ops.bartlett") from e
230
+
231
+ @ops_registry("keras.ops.batch_normalization")
232
+ class BatchNormalization(Lambda):
233
+ """Operation wrapping keras.ops.batch_normalization."""
234
+
235
+ def __init__(self, **kwargs):
236
+ try:
237
+ super().__init__(func=keras.ops.batch_normalization, **kwargs)
238
+ except AttributeError as e:
239
+ raise MissingKerasOps("BatchNormalization", "keras.ops.batch_normalization") from e
240
+
241
+ @ops_registry("keras.ops.bincount")
242
+ class Bincount(Lambda):
243
+ """Operation wrapping keras.ops.bincount."""
244
+
245
+ def __init__(self, **kwargs):
246
+ try:
247
+ super().__init__(func=keras.ops.bincount, **kwargs)
248
+ except AttributeError as e:
249
+ raise MissingKerasOps("Bincount", "keras.ops.bincount") from e
250
+
251
+ @ops_registry("keras.ops.bitwise_and")
252
+ class BitwiseAnd(Lambda):
253
+ """Operation wrapping keras.ops.bitwise_and."""
254
+
255
+ def __init__(self, **kwargs):
256
+ try:
257
+ super().__init__(func=keras.ops.bitwise_and, **kwargs)
258
+ except AttributeError as e:
259
+ raise MissingKerasOps("BitwiseAnd", "keras.ops.bitwise_and") from e
260
+
261
+ @ops_registry("keras.ops.bitwise_invert")
262
+ class BitwiseInvert(Lambda):
263
+ """Operation wrapping keras.ops.bitwise_invert."""
264
+
265
+ def __init__(self, **kwargs):
266
+ try:
267
+ super().__init__(func=keras.ops.bitwise_invert, **kwargs)
268
+ except AttributeError as e:
269
+ raise MissingKerasOps("BitwiseInvert", "keras.ops.bitwise_invert") from e
270
+
271
+ @ops_registry("keras.ops.bitwise_left_shift")
272
+ class BitwiseLeftShift(Lambda):
273
+ """Operation wrapping keras.ops.bitwise_left_shift."""
274
+
275
+ def __init__(self, **kwargs):
276
+ try:
277
+ super().__init__(func=keras.ops.bitwise_left_shift, **kwargs)
278
+ except AttributeError as e:
279
+ raise MissingKerasOps("BitwiseLeftShift", "keras.ops.bitwise_left_shift") from e
280
+
281
+ @ops_registry("keras.ops.bitwise_not")
282
+ class BitwiseNot(Lambda):
283
+ """Operation wrapping keras.ops.bitwise_not."""
284
+
285
+ def __init__(self, **kwargs):
286
+ try:
287
+ super().__init__(func=keras.ops.bitwise_not, **kwargs)
288
+ except AttributeError as e:
289
+ raise MissingKerasOps("BitwiseNot", "keras.ops.bitwise_not") from e
290
+
291
+ @ops_registry("keras.ops.bitwise_or")
292
+ class BitwiseOr(Lambda):
293
+ """Operation wrapping keras.ops.bitwise_or."""
294
+
295
+ def __init__(self, **kwargs):
296
+ try:
297
+ super().__init__(func=keras.ops.bitwise_or, **kwargs)
298
+ except AttributeError as e:
299
+ raise MissingKerasOps("BitwiseOr", "keras.ops.bitwise_or") from e
300
+
301
+ @ops_registry("keras.ops.bitwise_right_shift")
302
+ class BitwiseRightShift(Lambda):
303
+ """Operation wrapping keras.ops.bitwise_right_shift."""
304
+
305
+ def __init__(self, **kwargs):
306
+ try:
307
+ super().__init__(func=keras.ops.bitwise_right_shift, **kwargs)
308
+ except AttributeError as e:
309
+ raise MissingKerasOps("BitwiseRightShift", "keras.ops.bitwise_right_shift") from e
310
+
311
+ @ops_registry("keras.ops.bitwise_xor")
312
+ class BitwiseXor(Lambda):
313
+ """Operation wrapping keras.ops.bitwise_xor."""
314
+
315
+ def __init__(self, **kwargs):
316
+ try:
317
+ super().__init__(func=keras.ops.bitwise_xor, **kwargs)
318
+ except AttributeError as e:
319
+ raise MissingKerasOps("BitwiseXor", "keras.ops.bitwise_xor") from e
320
+
321
+ @ops_registry("keras.ops.blackman")
322
+ class Blackman(Lambda):
323
+ """Operation wrapping keras.ops.blackman."""
324
+
325
+ def __init__(self, **kwargs):
326
+ try:
327
+ super().__init__(func=keras.ops.blackman, **kwargs)
328
+ except AttributeError as e:
329
+ raise MissingKerasOps("Blackman", "keras.ops.blackman") from e
330
+
331
+ @ops_registry("keras.ops.broadcast_to")
332
+ class BroadcastTo(Lambda):
333
+ """Operation wrapping keras.ops.broadcast_to."""
334
+
335
+ def __init__(self, **kwargs):
336
+ try:
337
+ super().__init__(func=keras.ops.broadcast_to, **kwargs)
338
+ except AttributeError as e:
339
+ raise MissingKerasOps("BroadcastTo", "keras.ops.broadcast_to") from e
340
+
341
+ @ops_registry("keras.ops.cast")
342
+ class Cast(Lambda):
343
+ """Operation wrapping keras.ops.cast."""
344
+
345
+ def __init__(self, **kwargs):
346
+ try:
347
+ super().__init__(func=keras.ops.cast, **kwargs)
348
+ except AttributeError as e:
349
+ raise MissingKerasOps("Cast", "keras.ops.cast") from e
350
+
351
+ @ops_registry("keras.ops.cbrt")
352
+ class Cbrt(Lambda):
353
+ """Operation wrapping keras.ops.cbrt."""
354
+
355
+ def __init__(self, **kwargs):
356
+ try:
357
+ super().__init__(func=keras.ops.cbrt, **kwargs)
358
+ except AttributeError as e:
359
+ raise MissingKerasOps("Cbrt", "keras.ops.cbrt") from e
360
+
361
+ @ops_registry("keras.ops.ceil")
362
+ class Ceil(Lambda):
363
+ """Operation wrapping keras.ops.ceil."""
364
+
365
+ def __init__(self, **kwargs):
366
+ try:
367
+ super().__init__(func=keras.ops.ceil, **kwargs)
368
+ except AttributeError as e:
369
+ raise MissingKerasOps("Ceil", "keras.ops.ceil") from e
370
+
371
+ @ops_registry("keras.ops.celu")
372
+ class Celu(Lambda):
373
+ """Operation wrapping keras.ops.celu."""
374
+
375
+ def __init__(self, **kwargs):
376
+ try:
377
+ super().__init__(func=keras.ops.celu, **kwargs)
378
+ except AttributeError as e:
379
+ raise MissingKerasOps("Celu", "keras.ops.celu") from e
380
+
381
+ @ops_registry("keras.ops.cholesky")
382
+ class Cholesky(Lambda):
383
+ """Operation wrapping keras.ops.cholesky."""
384
+
385
+ def __init__(self, **kwargs):
386
+ try:
387
+ super().__init__(func=keras.ops.cholesky, **kwargs)
388
+ except AttributeError as e:
389
+ raise MissingKerasOps("Cholesky", "keras.ops.cholesky") from e
390
+
391
+ @ops_registry("keras.ops.clip")
392
+ class Clip(Lambda):
393
+ """Operation wrapping keras.ops.clip."""
394
+
395
+ def __init__(self, **kwargs):
396
+ try:
397
+ super().__init__(func=keras.ops.clip, **kwargs)
398
+ except AttributeError as e:
399
+ raise MissingKerasOps("Clip", "keras.ops.clip") from e
400
+
401
+ @ops_registry("keras.ops.conj")
402
+ class Conj(Lambda):
403
+ """Operation wrapping keras.ops.conj."""
404
+
405
+ def __init__(self, **kwargs):
406
+ try:
407
+ super().__init__(func=keras.ops.conj, **kwargs)
408
+ except AttributeError as e:
409
+ raise MissingKerasOps("Conj", "keras.ops.conj") from e
410
+
411
+ @ops_registry("keras.ops.conjugate")
412
+ class Conjugate(Lambda):
413
+ """Operation wrapping keras.ops.conjugate."""
414
+
415
+ def __init__(self, **kwargs):
416
+ try:
417
+ super().__init__(func=keras.ops.conjugate, **kwargs)
418
+ except AttributeError as e:
419
+ raise MissingKerasOps("Conjugate", "keras.ops.conjugate") from e
420
+
421
+ @ops_registry("keras.ops.convert_to_numpy")
422
+ class ConvertToNumpy(Lambda):
423
+ """Operation wrapping keras.ops.convert_to_numpy."""
424
+
425
+ def __init__(self, **kwargs):
426
+ try:
427
+ super().__init__(func=keras.ops.convert_to_numpy, **kwargs)
428
+ except AttributeError as e:
429
+ raise MissingKerasOps("ConvertToNumpy", "keras.ops.convert_to_numpy") from e
430
+
431
+ @ops_registry("keras.ops.convert_to_tensor")
432
+ class ConvertToTensor(Lambda):
433
+ """Operation wrapping keras.ops.convert_to_tensor."""
434
+
435
+ def __init__(self, **kwargs):
436
+ try:
437
+ super().__init__(func=keras.ops.convert_to_tensor, **kwargs)
438
+ except AttributeError as e:
439
+ raise MissingKerasOps("ConvertToTensor", "keras.ops.convert_to_tensor") from e
440
+
441
+ @ops_registry("keras.ops.copy")
442
+ class Copy(Lambda):
443
+ """Operation wrapping keras.ops.copy."""
444
+
445
+ def __init__(self, **kwargs):
446
+ try:
447
+ super().__init__(func=keras.ops.copy, **kwargs)
448
+ except AttributeError as e:
449
+ raise MissingKerasOps("Copy", "keras.ops.copy") from e
450
+
451
+ @ops_registry("keras.ops.corrcoef")
452
+ class Corrcoef(Lambda):
453
+ """Operation wrapping keras.ops.corrcoef."""
454
+
455
+ def __init__(self, **kwargs):
456
+ try:
457
+ super().__init__(func=keras.ops.corrcoef, **kwargs)
458
+ except AttributeError as e:
459
+ raise MissingKerasOps("Corrcoef", "keras.ops.corrcoef") from e
460
+
461
+ @ops_registry("keras.ops.cos")
462
+ class Cos(Lambda):
463
+ """Operation wrapping keras.ops.cos."""
464
+
465
+ def __init__(self, **kwargs):
466
+ try:
467
+ super().__init__(func=keras.ops.cos, **kwargs)
468
+ except AttributeError as e:
469
+ raise MissingKerasOps("Cos", "keras.ops.cos") from e
470
+
471
+ @ops_registry("keras.ops.cosh")
472
+ class Cosh(Lambda):
473
+ """Operation wrapping keras.ops.cosh."""
474
+
475
+ def __init__(self, **kwargs):
476
+ try:
477
+ super().__init__(func=keras.ops.cosh, **kwargs)
478
+ except AttributeError as e:
479
+ raise MissingKerasOps("Cosh", "keras.ops.cosh") from e
480
+
481
+ @ops_registry("keras.ops.count_nonzero")
482
+ class CountNonzero(Lambda):
483
+ """Operation wrapping keras.ops.count_nonzero."""
484
+
485
+ def __init__(self, **kwargs):
486
+ try:
487
+ super().__init__(func=keras.ops.count_nonzero, **kwargs)
488
+ except AttributeError as e:
489
+ raise MissingKerasOps("CountNonzero", "keras.ops.count_nonzero") from e
490
+
491
+ @ops_registry("keras.ops.cumprod")
492
+ class Cumprod(Lambda):
493
+ """Operation wrapping keras.ops.cumprod."""
494
+
495
+ def __init__(self, **kwargs):
496
+ try:
497
+ super().__init__(func=keras.ops.cumprod, **kwargs)
498
+ except AttributeError as e:
499
+ raise MissingKerasOps("Cumprod", "keras.ops.cumprod") from e
500
+
501
+ @ops_registry("keras.ops.cumsum")
502
+ class Cumsum(Lambda):
503
+ """Operation wrapping keras.ops.cumsum."""
504
+
505
+ def __init__(self, **kwargs):
506
+ try:
507
+ super().__init__(func=keras.ops.cumsum, **kwargs)
508
+ except AttributeError as e:
509
+ raise MissingKerasOps("Cumsum", "keras.ops.cumsum") from e
510
+
511
+ @ops_registry("keras.ops.deg2rad")
512
+ class Deg2rad(Lambda):
513
+ """Operation wrapping keras.ops.deg2rad."""
514
+
515
+ def __init__(self, **kwargs):
516
+ try:
517
+ super().__init__(func=keras.ops.deg2rad, **kwargs)
518
+ except AttributeError as e:
519
+ raise MissingKerasOps("Deg2rad", "keras.ops.deg2rad") from e
520
+
521
+ @ops_registry("keras.ops.det")
522
+ class Det(Lambda):
523
+ """Operation wrapping keras.ops.det."""
524
+
525
+ def __init__(self, **kwargs):
526
+ try:
527
+ super().__init__(func=keras.ops.det, **kwargs)
528
+ except AttributeError as e:
529
+ raise MissingKerasOps("Det", "keras.ops.det") from e
530
+
531
+ @ops_registry("keras.ops.diag")
532
+ class Diag(Lambda):
533
+ """Operation wrapping keras.ops.diag."""
534
+
535
+ def __init__(self, **kwargs):
536
+ try:
537
+ super().__init__(func=keras.ops.diag, **kwargs)
538
+ except AttributeError as e:
539
+ raise MissingKerasOps("Diag", "keras.ops.diag") from e
540
+
541
+ @ops_registry("keras.ops.diagflat")
542
+ class Diagflat(Lambda):
543
+ """Operation wrapping keras.ops.diagflat."""
544
+
545
+ def __init__(self, **kwargs):
546
+ try:
547
+ super().__init__(func=keras.ops.diagflat, **kwargs)
548
+ except AttributeError as e:
549
+ raise MissingKerasOps("Diagflat", "keras.ops.diagflat") from e
550
+
551
+ @ops_registry("keras.ops.diagonal")
552
+ class Diagonal(Lambda):
553
+ """Operation wrapping keras.ops.diagonal."""
554
+
555
+ def __init__(self, **kwargs):
556
+ try:
557
+ super().__init__(func=keras.ops.diagonal, **kwargs)
558
+ except AttributeError as e:
559
+ raise MissingKerasOps("Diagonal", "keras.ops.diagonal") from e
560
+
561
+ @ops_registry("keras.ops.digitize")
562
+ class Digitize(Lambda):
563
+ """Operation wrapping keras.ops.digitize."""
564
+
565
+ def __init__(self, **kwargs):
566
+ try:
567
+ super().__init__(func=keras.ops.digitize, **kwargs)
568
+ except AttributeError as e:
569
+ raise MissingKerasOps("Digitize", "keras.ops.digitize") from e
570
+
571
+ @ops_registry("keras.ops.dtype")
572
+ class Dtype(Lambda):
573
+ """Operation wrapping keras.ops.dtype."""
574
+
575
+ def __init__(self, **kwargs):
576
+ try:
577
+ super().__init__(func=keras.ops.dtype, **kwargs)
578
+ except AttributeError as e:
579
+ raise MissingKerasOps("Dtype", "keras.ops.dtype") from e
580
+
581
+ @ops_registry("keras.ops.eig")
582
+ class Eig(Lambda):
583
+ """Operation wrapping keras.ops.eig."""
584
+
585
+ def __init__(self, **kwargs):
586
+ try:
587
+ super().__init__(func=keras.ops.eig, **kwargs)
588
+ except AttributeError as e:
589
+ raise MissingKerasOps("Eig", "keras.ops.eig") from e
590
+
591
+ @ops_registry("keras.ops.eigh")
592
+ class Eigh(Lambda):
593
+ """Operation wrapping keras.ops.eigh."""
594
+
595
+ def __init__(self, **kwargs):
596
+ try:
597
+ super().__init__(func=keras.ops.eigh, **kwargs)
598
+ except AttributeError as e:
599
+ raise MissingKerasOps("Eigh", "keras.ops.eigh") from e
600
+
601
+ @ops_registry("keras.ops.elu")
602
+ class Elu(Lambda):
603
+ """Operation wrapping keras.ops.elu."""
604
+
605
+ def __init__(self, **kwargs):
606
+ try:
607
+ super().__init__(func=keras.ops.elu, **kwargs)
608
+ except AttributeError as e:
609
+ raise MissingKerasOps("Elu", "keras.ops.elu") from e
610
+
611
+ @ops_registry("keras.ops.erf")
612
+ class Erf(Lambda):
613
+ """Operation wrapping keras.ops.erf."""
614
+
615
+ def __init__(self, **kwargs):
616
+ try:
617
+ super().__init__(func=keras.ops.erf, **kwargs)
618
+ except AttributeError as e:
619
+ raise MissingKerasOps("Erf", "keras.ops.erf") from e
620
+
621
+ @ops_registry("keras.ops.erfinv")
622
+ class Erfinv(Lambda):
623
+ """Operation wrapping keras.ops.erfinv."""
624
+
625
+ def __init__(self, **kwargs):
626
+ try:
627
+ super().__init__(func=keras.ops.erfinv, **kwargs)
628
+ except AttributeError as e:
629
+ raise MissingKerasOps("Erfinv", "keras.ops.erfinv") from e
630
+
631
+ @ops_registry("keras.ops.exp")
632
+ class Exp(Lambda):
633
+ """Operation wrapping keras.ops.exp."""
634
+
635
+ def __init__(self, **kwargs):
636
+ try:
637
+ super().__init__(func=keras.ops.exp, **kwargs)
638
+ except AttributeError as e:
639
+ raise MissingKerasOps("Exp", "keras.ops.exp") from e
640
+
641
+ @ops_registry("keras.ops.exp2")
642
+ class Exp2(Lambda):
643
+ """Operation wrapping keras.ops.exp2."""
644
+
645
+ def __init__(self, **kwargs):
646
+ try:
647
+ super().__init__(func=keras.ops.exp2, **kwargs)
648
+ except AttributeError as e:
649
+ raise MissingKerasOps("Exp2", "keras.ops.exp2") from e
650
+
651
+ @ops_registry("keras.ops.expand_dims")
652
+ class ExpandDims(Lambda):
653
+ """Operation wrapping keras.ops.expand_dims."""
654
+
655
+ def __init__(self, **kwargs):
656
+ try:
657
+ super().__init__(func=keras.ops.expand_dims, **kwargs)
658
+ except AttributeError as e:
659
+ raise MissingKerasOps("ExpandDims", "keras.ops.expand_dims") from e
660
+
661
+ @ops_registry("keras.ops.expm1")
662
+ class Expm1(Lambda):
663
+ """Operation wrapping keras.ops.expm1."""
664
+
665
+ def __init__(self, **kwargs):
666
+ try:
667
+ super().__init__(func=keras.ops.expm1, **kwargs)
668
+ except AttributeError as e:
669
+ raise MissingKerasOps("Expm1", "keras.ops.expm1") from e
670
+
671
+ @ops_registry("keras.ops.extract_sequences")
672
+ class ExtractSequences(Lambda):
673
+ """Operation wrapping keras.ops.extract_sequences."""
674
+
675
+ def __init__(self, **kwargs):
676
+ try:
677
+ super().__init__(func=keras.ops.extract_sequences, **kwargs)
678
+ except AttributeError as e:
679
+ raise MissingKerasOps("ExtractSequences", "keras.ops.extract_sequences") from e
680
+
681
+ @ops_registry("keras.ops.fft")
682
+ class Fft(Lambda):
683
+ """Operation wrapping keras.ops.fft."""
684
+
685
+ def __init__(self, **kwargs):
686
+ try:
687
+ super().__init__(func=keras.ops.fft, **kwargs)
688
+ except AttributeError as e:
689
+ raise MissingKerasOps("Fft", "keras.ops.fft") from e
690
+
691
+ @ops_registry("keras.ops.fft2")
692
+ class Fft2(Lambda):
693
+ """Operation wrapping keras.ops.fft2."""
694
+
695
+ def __init__(self, **kwargs):
696
+ try:
697
+ super().__init__(func=keras.ops.fft2, **kwargs)
698
+ except AttributeError as e:
699
+ raise MissingKerasOps("Fft2", "keras.ops.fft2") from e
700
+
701
+ @ops_registry("keras.ops.flip")
702
+ class Flip(Lambda):
703
+ """Operation wrapping keras.ops.flip."""
704
+
705
+ def __init__(self, **kwargs):
706
+ try:
707
+ super().__init__(func=keras.ops.flip, **kwargs)
708
+ except AttributeError as e:
709
+ raise MissingKerasOps("Flip", "keras.ops.flip") from e
710
+
711
+ @ops_registry("keras.ops.floor")
712
+ class Floor(Lambda):
713
+ """Operation wrapping keras.ops.floor."""
714
+
715
+ def __init__(self, **kwargs):
716
+ try:
717
+ super().__init__(func=keras.ops.floor, **kwargs)
718
+ except AttributeError as e:
719
+ raise MissingKerasOps("Floor", "keras.ops.floor") from e
720
+
721
+ @ops_registry("keras.ops.full_like")
722
+ class FullLike(Lambda):
723
+ """Operation wrapping keras.ops.full_like."""
724
+
725
+ def __init__(self, **kwargs):
726
+ try:
727
+ super().__init__(func=keras.ops.full_like, **kwargs)
728
+ except AttributeError as e:
729
+ raise MissingKerasOps("FullLike", "keras.ops.full_like") from e
730
+
731
+ @ops_registry("keras.ops.gelu")
732
+ class Gelu(Lambda):
733
+ """Operation wrapping keras.ops.gelu."""
734
+
735
+ def __init__(self, **kwargs):
736
+ try:
737
+ super().__init__(func=keras.ops.gelu, **kwargs)
738
+ except AttributeError as e:
739
+ raise MissingKerasOps("Gelu", "keras.ops.gelu") from e
740
+
741
+ @ops_registry("keras.ops.get_item")
742
+ class GetItem(Lambda):
743
+ """Operation wrapping keras.ops.get_item."""
744
+
745
+ def __init__(self, **kwargs):
746
+ try:
747
+ super().__init__(func=keras.ops.get_item, **kwargs)
748
+ except AttributeError as e:
749
+ raise MissingKerasOps("GetItem", "keras.ops.get_item") from e
750
+
751
+ @ops_registry("keras.ops.glu")
752
+ class Glu(Lambda):
753
+ """Operation wrapping keras.ops.glu."""
754
+
755
+ def __init__(self, **kwargs):
756
+ try:
757
+ super().__init__(func=keras.ops.glu, **kwargs)
758
+ except AttributeError as e:
759
+ raise MissingKerasOps("Glu", "keras.ops.glu") from e
760
+
761
+ @ops_registry("keras.ops.hamming")
762
+ class Hamming(Lambda):
763
+ """Operation wrapping keras.ops.hamming."""
764
+
765
+ def __init__(self, **kwargs):
766
+ try:
767
+ super().__init__(func=keras.ops.hamming, **kwargs)
768
+ except AttributeError as e:
769
+ raise MissingKerasOps("Hamming", "keras.ops.hamming") from e
770
+
771
+ @ops_registry("keras.ops.hanning")
772
+ class Hanning(Lambda):
773
+ """Operation wrapping keras.ops.hanning."""
774
+
775
+ def __init__(self, **kwargs):
776
+ try:
777
+ super().__init__(func=keras.ops.hanning, **kwargs)
778
+ except AttributeError as e:
779
+ raise MissingKerasOps("Hanning", "keras.ops.hanning") from e
780
+
781
+ @ops_registry("keras.ops.hard_shrink")
782
+ class HardShrink(Lambda):
783
+ """Operation wrapping keras.ops.hard_shrink."""
784
+
785
+ def __init__(self, **kwargs):
786
+ try:
787
+ super().__init__(func=keras.ops.hard_shrink, **kwargs)
788
+ except AttributeError as e:
789
+ raise MissingKerasOps("HardShrink", "keras.ops.hard_shrink") from e
790
+
791
+ @ops_registry("keras.ops.hard_sigmoid")
792
+ class HardSigmoid(Lambda):
793
+ """Operation wrapping keras.ops.hard_sigmoid."""
794
+
795
+ def __init__(self, **kwargs):
796
+ try:
797
+ super().__init__(func=keras.ops.hard_sigmoid, **kwargs)
798
+ except AttributeError as e:
799
+ raise MissingKerasOps("HardSigmoid", "keras.ops.hard_sigmoid") from e
800
+
801
+ @ops_registry("keras.ops.hard_silu")
802
+ class HardSilu(Lambda):
803
+ """Operation wrapping keras.ops.hard_silu."""
804
+
805
+ def __init__(self, **kwargs):
806
+ try:
807
+ super().__init__(func=keras.ops.hard_silu, **kwargs)
808
+ except AttributeError as e:
809
+ raise MissingKerasOps("HardSilu", "keras.ops.hard_silu") from e
810
+
811
+ @ops_registry("keras.ops.hard_swish")
812
+ class HardSwish(Lambda):
813
+ """Operation wrapping keras.ops.hard_swish."""
814
+
815
+ def __init__(self, **kwargs):
816
+ try:
817
+ super().__init__(func=keras.ops.hard_swish, **kwargs)
818
+ except AttributeError as e:
819
+ raise MissingKerasOps("HardSwish", "keras.ops.hard_swish") from e
820
+
821
+ @ops_registry("keras.ops.hard_tanh")
822
+ class HardTanh(Lambda):
823
+ """Operation wrapping keras.ops.hard_tanh."""
824
+
825
+ def __init__(self, **kwargs):
826
+ try:
827
+ super().__init__(func=keras.ops.hard_tanh, **kwargs)
828
+ except AttributeError as e:
829
+ raise MissingKerasOps("HardTanh", "keras.ops.hard_tanh") from e
830
+
831
+ @ops_registry("keras.ops.histogram")
832
+ class Histogram(Lambda):
833
+ """Operation wrapping keras.ops.histogram."""
834
+
835
+ def __init__(self, **kwargs):
836
+ try:
837
+ super().__init__(func=keras.ops.histogram, **kwargs)
838
+ except AttributeError as e:
839
+ raise MissingKerasOps("Histogram", "keras.ops.histogram") from e
840
+
841
+ @ops_registry("keras.ops.ifft2")
842
+ class Ifft2(Lambda):
843
+ """Operation wrapping keras.ops.ifft2."""
844
+
845
+ def __init__(self, **kwargs):
846
+ try:
847
+ super().__init__(func=keras.ops.ifft2, **kwargs)
848
+ except AttributeError as e:
849
+ raise MissingKerasOps("Ifft2", "keras.ops.ifft2") from e
850
+
851
+ @ops_registry("keras.ops.imag")
852
+ class Imag(Lambda):
853
+ """Operation wrapping keras.ops.imag."""
854
+
855
+ def __init__(self, **kwargs):
856
+ try:
857
+ super().__init__(func=keras.ops.imag, **kwargs)
858
+ except AttributeError as e:
859
+ raise MissingKerasOps("Imag", "keras.ops.imag") from e
860
+
861
+ @ops_registry("keras.ops.inv")
862
+ class Inv(Lambda):
863
+ """Operation wrapping keras.ops.inv."""
864
+
865
+ def __init__(self, **kwargs):
866
+ try:
867
+ super().__init__(func=keras.ops.inv, **kwargs)
868
+ except AttributeError as e:
869
+ raise MissingKerasOps("Inv", "keras.ops.inv") from e
870
+
871
+ @ops_registry("keras.ops.irfft")
872
+ class Irfft(Lambda):
873
+ """Operation wrapping keras.ops.irfft."""
874
+
875
+ def __init__(self, **kwargs):
876
+ try:
877
+ super().__init__(func=keras.ops.irfft, **kwargs)
878
+ except AttributeError as e:
879
+ raise MissingKerasOps("Irfft", "keras.ops.irfft") from e
880
+
881
+ @ops_registry("keras.ops.is_tensor")
882
+ class IsTensor(Lambda):
883
+ """Operation wrapping keras.ops.is_tensor."""
884
+
885
+ def __init__(self, **kwargs):
886
+ try:
887
+ super().__init__(func=keras.ops.is_tensor, **kwargs)
888
+ except AttributeError as e:
889
+ raise MissingKerasOps("IsTensor", "keras.ops.is_tensor") from e
890
+
891
+ @ops_registry("keras.ops.isfinite")
892
+ class Isfinite(Lambda):
893
+ """Operation wrapping keras.ops.isfinite."""
894
+
895
+ def __init__(self, **kwargs):
896
+ try:
897
+ super().__init__(func=keras.ops.isfinite, **kwargs)
898
+ except AttributeError as e:
899
+ raise MissingKerasOps("Isfinite", "keras.ops.isfinite") from e
900
+
901
+ @ops_registry("keras.ops.isinf")
902
+ class Isinf(Lambda):
903
+ """Operation wrapping keras.ops.isinf."""
904
+
905
+ def __init__(self, **kwargs):
906
+ try:
907
+ super().__init__(func=keras.ops.isinf, **kwargs)
908
+ except AttributeError as e:
909
+ raise MissingKerasOps("Isinf", "keras.ops.isinf") from e
910
+
911
+ @ops_registry("keras.ops.isnan")
912
+ class Isnan(Lambda):
913
+ """Operation wrapping keras.ops.isnan."""
914
+
915
+ def __init__(self, **kwargs):
916
+ try:
917
+ super().__init__(func=keras.ops.isnan, **kwargs)
918
+ except AttributeError as e:
919
+ raise MissingKerasOps("Isnan", "keras.ops.isnan") from e
920
+
921
+ @ops_registry("keras.ops.istft")
922
+ class Istft(Lambda):
923
+ """Operation wrapping keras.ops.istft."""
924
+
925
+ def __init__(self, **kwargs):
926
+ try:
927
+ super().__init__(func=keras.ops.istft, **kwargs)
928
+ except AttributeError as e:
929
+ raise MissingKerasOps("Istft", "keras.ops.istft") from e
930
+
931
+ @ops_registry("keras.ops.kaiser")
932
+ class Kaiser(Lambda):
933
+ """Operation wrapping keras.ops.kaiser."""
934
+
935
+ def __init__(self, **kwargs):
936
+ try:
937
+ super().__init__(func=keras.ops.kaiser, **kwargs)
938
+ except AttributeError as e:
939
+ raise MissingKerasOps("Kaiser", "keras.ops.kaiser") from e
940
+
941
+ @ops_registry("keras.ops.layer_normalization")
942
+ class LayerNormalization(Lambda):
943
+ """Operation wrapping keras.ops.layer_normalization."""
944
+
945
+ def __init__(self, **kwargs):
946
+ try:
947
+ super().__init__(func=keras.ops.layer_normalization, **kwargs)
948
+ except AttributeError as e:
949
+ raise MissingKerasOps("LayerNormalization", "keras.ops.layer_normalization") from e
950
+
951
+ @ops_registry("keras.ops.leaky_relu")
952
+ class LeakyRelu(Lambda):
953
+ """Operation wrapping keras.ops.leaky_relu."""
954
+
955
+ def __init__(self, **kwargs):
956
+ try:
957
+ super().__init__(func=keras.ops.leaky_relu, **kwargs)
958
+ except AttributeError as e:
959
+ raise MissingKerasOps("LeakyRelu", "keras.ops.leaky_relu") from e
960
+
961
+ @ops_registry("keras.ops.left_shift")
962
+ class LeftShift(Lambda):
963
+ """Operation wrapping keras.ops.left_shift."""
964
+
965
+ def __init__(self, **kwargs):
966
+ try:
967
+ super().__init__(func=keras.ops.left_shift, **kwargs)
968
+ except AttributeError as e:
969
+ raise MissingKerasOps("LeftShift", "keras.ops.left_shift") from e
970
+
971
+ @ops_registry("keras.ops.log")
972
+ class Log(Lambda):
973
+ """Operation wrapping keras.ops.log."""
974
+
975
+ def __init__(self, **kwargs):
976
+ try:
977
+ super().__init__(func=keras.ops.log, **kwargs)
978
+ except AttributeError as e:
979
+ raise MissingKerasOps("Log", "keras.ops.log") from e
980
+
981
+ @ops_registry("keras.ops.log10")
982
+ class Log10(Lambda):
983
+ """Operation wrapping keras.ops.log10."""
984
+
985
+ def __init__(self, **kwargs):
986
+ try:
987
+ super().__init__(func=keras.ops.log10, **kwargs)
988
+ except AttributeError as e:
989
+ raise MissingKerasOps("Log10", "keras.ops.log10") from e
990
+
991
+ @ops_registry("keras.ops.log1p")
992
+ class Log1p(Lambda):
993
+ """Operation wrapping keras.ops.log1p."""
994
+
995
+ def __init__(self, **kwargs):
996
+ try:
997
+ super().__init__(func=keras.ops.log1p, **kwargs)
998
+ except AttributeError as e:
999
+ raise MissingKerasOps("Log1p", "keras.ops.log1p") from e
1000
+
1001
+ @ops_registry("keras.ops.log2")
1002
+ class Log2(Lambda):
1003
+ """Operation wrapping keras.ops.log2."""
1004
+
1005
+ def __init__(self, **kwargs):
1006
+ try:
1007
+ super().__init__(func=keras.ops.log2, **kwargs)
1008
+ except AttributeError as e:
1009
+ raise MissingKerasOps("Log2", "keras.ops.log2") from e
1010
+
1011
+ @ops_registry("keras.ops.log_sigmoid")
1012
+ class LogSigmoid(Lambda):
1013
+ """Operation wrapping keras.ops.log_sigmoid."""
1014
+
1015
+ def __init__(self, **kwargs):
1016
+ try:
1017
+ super().__init__(func=keras.ops.log_sigmoid, **kwargs)
1018
+ except AttributeError as e:
1019
+ raise MissingKerasOps("LogSigmoid", "keras.ops.log_sigmoid") from e
1020
+
1021
+ @ops_registry("keras.ops.log_softmax")
1022
+ class LogSoftmax(Lambda):
1023
+ """Operation wrapping keras.ops.log_softmax."""
1024
+
1025
+ def __init__(self, **kwargs):
1026
+ try:
1027
+ super().__init__(func=keras.ops.log_softmax, **kwargs)
1028
+ except AttributeError as e:
1029
+ raise MissingKerasOps("LogSoftmax", "keras.ops.log_softmax") from e
1030
+
1031
+ @ops_registry("keras.ops.logdet")
1032
+ class Logdet(Lambda):
1033
+ """Operation wrapping keras.ops.logdet."""
1034
+
1035
+ def __init__(self, **kwargs):
1036
+ try:
1037
+ super().__init__(func=keras.ops.logdet, **kwargs)
1038
+ except AttributeError as e:
1039
+ raise MissingKerasOps("Logdet", "keras.ops.logdet") from e
1040
+
1041
+ @ops_registry("keras.ops.logical_not")
1042
+ class LogicalNot(Lambda):
1043
+ """Operation wrapping keras.ops.logical_not."""
1044
+
1045
+ def __init__(self, **kwargs):
1046
+ try:
1047
+ super().__init__(func=keras.ops.logical_not, **kwargs)
1048
+ except AttributeError as e:
1049
+ raise MissingKerasOps("LogicalNot", "keras.ops.logical_not") from e
1050
+
1051
+ @ops_registry("keras.ops.logsumexp")
1052
+ class Logsumexp(Lambda):
1053
+ """Operation wrapping keras.ops.logsumexp."""
1054
+
1055
+ def __init__(self, **kwargs):
1056
+ try:
1057
+ super().__init__(func=keras.ops.logsumexp, **kwargs)
1058
+ except AttributeError as e:
1059
+ raise MissingKerasOps("Logsumexp", "keras.ops.logsumexp") from e
1060
+
1061
+ @ops_registry("keras.ops.lu_factor")
1062
+ class LuFactor(Lambda):
1063
+ """Operation wrapping keras.ops.lu_factor."""
1064
+
1065
+ def __init__(self, **kwargs):
1066
+ try:
1067
+ super().__init__(func=keras.ops.lu_factor, **kwargs)
1068
+ except AttributeError as e:
1069
+ raise MissingKerasOps("LuFactor", "keras.ops.lu_factor") from e
1070
+
1071
+ @ops_registry("keras.ops.max")
1072
+ class Max(Lambda):
1073
+ """Operation wrapping keras.ops.max."""
1074
+
1075
+ def __init__(self, **kwargs):
1076
+ try:
1077
+ super().__init__(func=keras.ops.max, **kwargs)
1078
+ except AttributeError as e:
1079
+ raise MissingKerasOps("Max", "keras.ops.max") from e
1080
+
1081
+ @ops_registry("keras.ops.mean")
1082
+ class Mean(Lambda):
1083
+ """Operation wrapping keras.ops.mean."""
1084
+
1085
+ def __init__(self, **kwargs):
1086
+ try:
1087
+ super().__init__(func=keras.ops.mean, **kwargs)
1088
+ except AttributeError as e:
1089
+ raise MissingKerasOps("Mean", "keras.ops.mean") from e
1090
+
1091
+ @ops_registry("keras.ops.median")
1092
+ class Median(Lambda):
1093
+ """Operation wrapping keras.ops.median."""
1094
+
1095
+ def __init__(self, **kwargs):
1096
+ try:
1097
+ super().__init__(func=keras.ops.median, **kwargs)
1098
+ except AttributeError as e:
1099
+ raise MissingKerasOps("Median", "keras.ops.median") from e
1100
+
1101
+ @ops_registry("keras.ops.meshgrid")
1102
+ class Meshgrid(Lambda):
1103
+ """Operation wrapping keras.ops.meshgrid."""
1104
+
1105
+ def __init__(self, **kwargs):
1106
+ try:
1107
+ super().__init__(func=keras.ops.meshgrid, **kwargs)
1108
+ except AttributeError as e:
1109
+ raise MissingKerasOps("Meshgrid", "keras.ops.meshgrid") from e
1110
+
1111
+ @ops_registry("keras.ops.min")
1112
+ class Min(Lambda):
1113
+ """Operation wrapping keras.ops.min."""
1114
+
1115
+ def __init__(self, **kwargs):
1116
+ try:
1117
+ super().__init__(func=keras.ops.min, **kwargs)
1118
+ except AttributeError as e:
1119
+ raise MissingKerasOps("Min", "keras.ops.min") from e
1120
+
1121
+ @ops_registry("keras.ops.moments")
1122
+ class Moments(Lambda):
1123
+ """Operation wrapping keras.ops.moments."""
1124
+
1125
+ def __init__(self, **kwargs):
1126
+ try:
1127
+ super().__init__(func=keras.ops.moments, **kwargs)
1128
+ except AttributeError as e:
1129
+ raise MissingKerasOps("Moments", "keras.ops.moments") from e
1130
+
1131
+ @ops_registry("keras.ops.moveaxis")
1132
+ class Moveaxis(Lambda):
1133
+ """Operation wrapping keras.ops.moveaxis."""
1134
+
1135
+ def __init__(self, **kwargs):
1136
+ try:
1137
+ super().__init__(func=keras.ops.moveaxis, **kwargs)
1138
+ except AttributeError as e:
1139
+ raise MissingKerasOps("Moveaxis", "keras.ops.moveaxis") from e
1140
+
1141
+ @ops_registry("keras.ops.nan_to_num")
1142
+ class NanToNum(Lambda):
1143
+ """Operation wrapping keras.ops.nan_to_num."""
1144
+
1145
+ def __init__(self, **kwargs):
1146
+ try:
1147
+ super().__init__(func=keras.ops.nan_to_num, **kwargs)
1148
+ except AttributeError as e:
1149
+ raise MissingKerasOps("NanToNum", "keras.ops.nan_to_num") from e
1150
+
1151
+ @ops_registry("keras.ops.ndim")
1152
+ class Ndim(Lambda):
1153
+ """Operation wrapping keras.ops.ndim."""
1154
+
1155
+ def __init__(self, **kwargs):
1156
+ try:
1157
+ super().__init__(func=keras.ops.ndim, **kwargs)
1158
+ except AttributeError as e:
1159
+ raise MissingKerasOps("Ndim", "keras.ops.ndim") from e
1160
+
1161
+ @ops_registry("keras.ops.negative")
1162
+ class Negative(Lambda):
1163
+ """Operation wrapping keras.ops.negative."""
1164
+
1165
+ def __init__(self, **kwargs):
1166
+ try:
1167
+ super().__init__(func=keras.ops.negative, **kwargs)
1168
+ except AttributeError as e:
1169
+ raise MissingKerasOps("Negative", "keras.ops.negative") from e
1170
+
1171
+ @ops_registry("keras.ops.nonzero")
1172
+ class Nonzero(Lambda):
1173
+ """Operation wrapping keras.ops.nonzero."""
1174
+
1175
+ def __init__(self, **kwargs):
1176
+ try:
1177
+ super().__init__(func=keras.ops.nonzero, **kwargs)
1178
+ except AttributeError as e:
1179
+ raise MissingKerasOps("Nonzero", "keras.ops.nonzero") from e
1180
+
1181
+ @ops_registry("keras.ops.norm")
1182
+ class Norm(Lambda):
1183
+ """Operation wrapping keras.ops.norm."""
1184
+
1185
+ def __init__(self, **kwargs):
1186
+ try:
1187
+ super().__init__(func=keras.ops.norm, **kwargs)
1188
+ except AttributeError as e:
1189
+ raise MissingKerasOps("Norm", "keras.ops.norm") from e
1190
+
1191
+ @ops_registry("keras.ops.normalize")
1192
+ class Normalize(Lambda):
1193
+ """Operation wrapping keras.ops.normalize."""
1194
+
1195
+ def __init__(self, **kwargs):
1196
+ try:
1197
+ super().__init__(func=keras.ops.normalize, **kwargs)
1198
+ except AttributeError as e:
1199
+ raise MissingKerasOps("Normalize", "keras.ops.normalize") from e
1200
+
1201
+ @ops_registry("keras.ops.one_hot")
1202
+ class OneHot(Lambda):
1203
+ """Operation wrapping keras.ops.one_hot."""
1204
+
1205
+ def __init__(self, **kwargs):
1206
+ try:
1207
+ super().__init__(func=keras.ops.one_hot, **kwargs)
1208
+ except AttributeError as e:
1209
+ raise MissingKerasOps("OneHot", "keras.ops.one_hot") from e
1210
+
1211
+ @ops_registry("keras.ops.ones_like")
1212
+ class OnesLike(Lambda):
1213
+ """Operation wrapping keras.ops.ones_like."""
1214
+
1215
+ def __init__(self, **kwargs):
1216
+ try:
1217
+ super().__init__(func=keras.ops.ones_like, **kwargs)
1218
+ except AttributeError as e:
1219
+ raise MissingKerasOps("OnesLike", "keras.ops.ones_like") from e
1220
+
1221
+ @ops_registry("keras.ops.pad")
1222
+ class Pad(Lambda):
1223
+ """Operation wrapping keras.ops.pad."""
1224
+
1225
+ def __init__(self, **kwargs):
1226
+ try:
1227
+ super().__init__(func=keras.ops.pad, **kwargs)
1228
+ except AttributeError as e:
1229
+ raise MissingKerasOps("Pad", "keras.ops.pad") from e
1230
+
1231
+ @ops_registry("keras.ops.prod")
1232
+ class Prod(Lambda):
1233
+ """Operation wrapping keras.ops.prod."""
1234
+
1235
+ def __init__(self, **kwargs):
1236
+ try:
1237
+ super().__init__(func=keras.ops.prod, **kwargs)
1238
+ except AttributeError as e:
1239
+ raise MissingKerasOps("Prod", "keras.ops.prod") from e
1240
+
1241
+ @ops_registry("keras.ops.qr")
1242
+ class Qr(Lambda):
1243
+ """Operation wrapping keras.ops.qr."""
1244
+
1245
+ def __init__(self, **kwargs):
1246
+ try:
1247
+ super().__init__(func=keras.ops.qr, **kwargs)
1248
+ except AttributeError as e:
1249
+ raise MissingKerasOps("Qr", "keras.ops.qr") from e
1250
+
1251
+ @ops_registry("keras.ops.quantile")
1252
+ class Quantile(Lambda):
1253
+ """Operation wrapping keras.ops.quantile."""
1254
+
1255
+ def __init__(self, **kwargs):
1256
+ try:
1257
+ super().__init__(func=keras.ops.quantile, **kwargs)
1258
+ except AttributeError as e:
1259
+ raise MissingKerasOps("Quantile", "keras.ops.quantile") from e
1260
+
1261
+ @ops_registry("keras.ops.ravel")
1262
+ class Ravel(Lambda):
1263
+ """Operation wrapping keras.ops.ravel."""
1264
+
1265
+ def __init__(self, **kwargs):
1266
+ try:
1267
+ super().__init__(func=keras.ops.ravel, **kwargs)
1268
+ except AttributeError as e:
1269
+ raise MissingKerasOps("Ravel", "keras.ops.ravel") from e
1270
+
1271
+ @ops_registry("keras.ops.real")
1272
+ class Real(Lambda):
1273
+ """Operation wrapping keras.ops.real."""
1274
+
1275
+ def __init__(self, **kwargs):
1276
+ try:
1277
+ super().__init__(func=keras.ops.real, **kwargs)
1278
+ except AttributeError as e:
1279
+ raise MissingKerasOps("Real", "keras.ops.real") from e
1280
+
1281
+ @ops_registry("keras.ops.reciprocal")
1282
+ class Reciprocal(Lambda):
1283
+ """Operation wrapping keras.ops.reciprocal."""
1284
+
1285
+ def __init__(self, **kwargs):
1286
+ try:
1287
+ super().__init__(func=keras.ops.reciprocal, **kwargs)
1288
+ except AttributeError as e:
1289
+ raise MissingKerasOps("Reciprocal", "keras.ops.reciprocal") from e
1290
+
1291
+ @ops_registry("keras.ops.relu")
1292
+ class Relu(Lambda):
1293
+ """Operation wrapping keras.ops.relu."""
1294
+
1295
+ def __init__(self, **kwargs):
1296
+ try:
1297
+ super().__init__(func=keras.ops.relu, **kwargs)
1298
+ except AttributeError as e:
1299
+ raise MissingKerasOps("Relu", "keras.ops.relu") from e
1300
+
1301
+ @ops_registry("keras.ops.relu6")
1302
+ class Relu6(Lambda):
1303
+ """Operation wrapping keras.ops.relu6."""
1304
+
1305
+ def __init__(self, **kwargs):
1306
+ try:
1307
+ super().__init__(func=keras.ops.relu6, **kwargs)
1308
+ except AttributeError as e:
1309
+ raise MissingKerasOps("Relu6", "keras.ops.relu6") from e
1310
+
1311
+ @ops_registry("keras.ops.repeat")
1312
+ class Repeat(Lambda):
1313
+ """Operation wrapping keras.ops.repeat."""
1314
+
1315
+ def __init__(self, **kwargs):
1316
+ try:
1317
+ super().__init__(func=keras.ops.repeat, **kwargs)
1318
+ except AttributeError as e:
1319
+ raise MissingKerasOps("Repeat", "keras.ops.repeat") from e
1320
+
1321
+ @ops_registry("keras.ops.reshape")
1322
+ class Reshape(Lambda):
1323
+ """Operation wrapping keras.ops.reshape."""
1324
+
1325
+ def __init__(self, **kwargs):
1326
+ try:
1327
+ super().__init__(func=keras.ops.reshape, **kwargs)
1328
+ except AttributeError as e:
1329
+ raise MissingKerasOps("Reshape", "keras.ops.reshape") from e
1330
+
1331
+ @ops_registry("keras.ops.rfft")
1332
+ class Rfft(Lambda):
1333
+ """Operation wrapping keras.ops.rfft."""
1334
+
1335
+ def __init__(self, **kwargs):
1336
+ try:
1337
+ super().__init__(func=keras.ops.rfft, **kwargs)
1338
+ except AttributeError as e:
1339
+ raise MissingKerasOps("Rfft", "keras.ops.rfft") from e
1340
+
1341
+ @ops_registry("keras.ops.right_shift")
1342
+ class RightShift(Lambda):
1343
+ """Operation wrapping keras.ops.right_shift."""
1344
+
1345
+ def __init__(self, **kwargs):
1346
+ try:
1347
+ super().__init__(func=keras.ops.right_shift, **kwargs)
1348
+ except AttributeError as e:
1349
+ raise MissingKerasOps("RightShift", "keras.ops.right_shift") from e
1350
+
1351
+ @ops_registry("keras.ops.rms_normalization")
1352
+ class RmsNormalization(Lambda):
1353
+ """Operation wrapping keras.ops.rms_normalization."""
1354
+
1355
+ def __init__(self, **kwargs):
1356
+ try:
1357
+ super().__init__(func=keras.ops.rms_normalization, **kwargs)
1358
+ except AttributeError as e:
1359
+ raise MissingKerasOps("RmsNormalization", "keras.ops.rms_normalization") from e
1360
+
1361
+ @ops_registry("keras.ops.roll")
1362
+ class Roll(Lambda):
1363
+ """Operation wrapping keras.ops.roll."""
1364
+
1365
+ def __init__(self, **kwargs):
1366
+ try:
1367
+ super().__init__(func=keras.ops.roll, **kwargs)
1368
+ except AttributeError as e:
1369
+ raise MissingKerasOps("Roll", "keras.ops.roll") from e
1370
+
1371
+ @ops_registry("keras.ops.round")
1372
+ class Round(Lambda):
1373
+ """Operation wrapping keras.ops.round."""
1374
+
1375
+ def __init__(self, **kwargs):
1376
+ try:
1377
+ super().__init__(func=keras.ops.round, **kwargs)
1378
+ except AttributeError as e:
1379
+ raise MissingKerasOps("Round", "keras.ops.round") from e
1380
+
1381
+ @ops_registry("keras.ops.rsqrt")
1382
+ class Rsqrt(Lambda):
1383
+ """Operation wrapping keras.ops.rsqrt."""
1384
+
1385
+ def __init__(self, **kwargs):
1386
+ try:
1387
+ super().__init__(func=keras.ops.rsqrt, **kwargs)
1388
+ except AttributeError as e:
1389
+ raise MissingKerasOps("Rsqrt", "keras.ops.rsqrt") from e
1390
+
1391
+ @ops_registry("keras.ops.saturate_cast")
1392
+ class SaturateCast(Lambda):
1393
+ """Operation wrapping keras.ops.saturate_cast."""
1394
+
1395
+ def __init__(self, **kwargs):
1396
+ try:
1397
+ super().__init__(func=keras.ops.saturate_cast, **kwargs)
1398
+ except AttributeError as e:
1399
+ raise MissingKerasOps("SaturateCast", "keras.ops.saturate_cast") from e
1400
+
1401
+ @ops_registry("keras.ops.selu")
1402
+ class Selu(Lambda):
1403
+ """Operation wrapping keras.ops.selu."""
1404
+
1405
+ def __init__(self, **kwargs):
1406
+ try:
1407
+ super().__init__(func=keras.ops.selu, **kwargs)
1408
+ except AttributeError as e:
1409
+ raise MissingKerasOps("Selu", "keras.ops.selu") from e
1410
+
1411
+ @ops_registry("keras.ops.shape")
1412
+ class Shape(Lambda):
1413
+ """Operation wrapping keras.ops.shape."""
1414
+
1415
+ def __init__(self, **kwargs):
1416
+ try:
1417
+ super().__init__(func=keras.ops.shape, **kwargs)
1418
+ except AttributeError as e:
1419
+ raise MissingKerasOps("Shape", "keras.ops.shape") from e
1420
+
1421
+ @ops_registry("keras.ops.sigmoid")
1422
+ class Sigmoid(Lambda):
1423
+ """Operation wrapping keras.ops.sigmoid."""
1424
+
1425
+ def __init__(self, **kwargs):
1426
+ try:
1427
+ super().__init__(func=keras.ops.sigmoid, **kwargs)
1428
+ except AttributeError as e:
1429
+ raise MissingKerasOps("Sigmoid", "keras.ops.sigmoid") from e
1430
+
1431
+ @ops_registry("keras.ops.sign")
1432
+ class Sign(Lambda):
1433
+ """Operation wrapping keras.ops.sign."""
1434
+
1435
+ def __init__(self, **kwargs):
1436
+ try:
1437
+ super().__init__(func=keras.ops.sign, **kwargs)
1438
+ except AttributeError as e:
1439
+ raise MissingKerasOps("Sign", "keras.ops.sign") from e
1440
+
1441
+ @ops_registry("keras.ops.signbit")
1442
+ class Signbit(Lambda):
1443
+ """Operation wrapping keras.ops.signbit."""
1444
+
1445
+ def __init__(self, **kwargs):
1446
+ try:
1447
+ super().__init__(func=keras.ops.signbit, **kwargs)
1448
+ except AttributeError as e:
1449
+ raise MissingKerasOps("Signbit", "keras.ops.signbit") from e
1450
+
1451
+ @ops_registry("keras.ops.silu")
1452
+ class Silu(Lambda):
1453
+ """Operation wrapping keras.ops.silu."""
1454
+
1455
+ def __init__(self, **kwargs):
1456
+ try:
1457
+ super().__init__(func=keras.ops.silu, **kwargs)
1458
+ except AttributeError as e:
1459
+ raise MissingKerasOps("Silu", "keras.ops.silu") from e
1460
+
1461
+ @ops_registry("keras.ops.sin")
1462
+ class Sin(Lambda):
1463
+ """Operation wrapping keras.ops.sin."""
1464
+
1465
+ def __init__(self, **kwargs):
1466
+ try:
1467
+ super().__init__(func=keras.ops.sin, **kwargs)
1468
+ except AttributeError as e:
1469
+ raise MissingKerasOps("Sin", "keras.ops.sin") from e
1470
+
1471
+ @ops_registry("keras.ops.sinh")
1472
+ class Sinh(Lambda):
1473
+ """Operation wrapping keras.ops.sinh."""
1474
+
1475
+ def __init__(self, **kwargs):
1476
+ try:
1477
+ super().__init__(func=keras.ops.sinh, **kwargs)
1478
+ except AttributeError as e:
1479
+ raise MissingKerasOps("Sinh", "keras.ops.sinh") from e
1480
+
1481
+ @ops_registry("keras.ops.size")
1482
+ class Size(Lambda):
1483
+ """Operation wrapping keras.ops.size."""
1484
+
1485
+ def __init__(self, **kwargs):
1486
+ try:
1487
+ super().__init__(func=keras.ops.size, **kwargs)
1488
+ except AttributeError as e:
1489
+ raise MissingKerasOps("Size", "keras.ops.size") from e
1490
+
1491
+ @ops_registry("keras.ops.slogdet")
1492
+ class Slogdet(Lambda):
1493
+ """Operation wrapping keras.ops.slogdet."""
1494
+
1495
+ def __init__(self, **kwargs):
1496
+ try:
1497
+ super().__init__(func=keras.ops.slogdet, **kwargs)
1498
+ except AttributeError as e:
1499
+ raise MissingKerasOps("Slogdet", "keras.ops.slogdet") from e
1500
+
1501
+ @ops_registry("keras.ops.soft_shrink")
1502
+ class SoftShrink(Lambda):
1503
+ """Operation wrapping keras.ops.soft_shrink."""
1504
+
1505
+ def __init__(self, **kwargs):
1506
+ try:
1507
+ super().__init__(func=keras.ops.soft_shrink, **kwargs)
1508
+ except AttributeError as e:
1509
+ raise MissingKerasOps("SoftShrink", "keras.ops.soft_shrink") from e
1510
+
1511
+ @ops_registry("keras.ops.softmax")
1512
+ class Softmax(Lambda):
1513
+ """Operation wrapping keras.ops.softmax."""
1514
+
1515
+ def __init__(self, **kwargs):
1516
+ try:
1517
+ super().__init__(func=keras.ops.softmax, **kwargs)
1518
+ except AttributeError as e:
1519
+ raise MissingKerasOps("Softmax", "keras.ops.softmax") from e
1520
+
1521
+ @ops_registry("keras.ops.softplus")
1522
+ class Softplus(Lambda):
1523
+ """Operation wrapping keras.ops.softplus."""
1524
+
1525
+ def __init__(self, **kwargs):
1526
+ try:
1527
+ super().__init__(func=keras.ops.softplus, **kwargs)
1528
+ except AttributeError as e:
1529
+ raise MissingKerasOps("Softplus", "keras.ops.softplus") from e
1530
+
1531
+ @ops_registry("keras.ops.softsign")
1532
+ class Softsign(Lambda):
1533
+ """Operation wrapping keras.ops.softsign."""
1534
+
1535
+ def __init__(self, **kwargs):
1536
+ try:
1537
+ super().__init__(func=keras.ops.softsign, **kwargs)
1538
+ except AttributeError as e:
1539
+ raise MissingKerasOps("Softsign", "keras.ops.softsign") from e
1540
+
1541
+ @ops_registry("keras.ops.sort")
1542
+ class Sort(Lambda):
1543
+ """Operation wrapping keras.ops.sort."""
1544
+
1545
+ def __init__(self, **kwargs):
1546
+ try:
1547
+ super().__init__(func=keras.ops.sort, **kwargs)
1548
+ except AttributeError as e:
1549
+ raise MissingKerasOps("Sort", "keras.ops.sort") from e
1550
+
1551
+ @ops_registry("keras.ops.sparse_plus")
1552
+ class SparsePlus(Lambda):
1553
+ """Operation wrapping keras.ops.sparse_plus."""
1554
+
1555
+ def __init__(self, **kwargs):
1556
+ try:
1557
+ super().__init__(func=keras.ops.sparse_plus, **kwargs)
1558
+ except AttributeError as e:
1559
+ raise MissingKerasOps("SparsePlus", "keras.ops.sparse_plus") from e
1560
+
1561
+ @ops_registry("keras.ops.sparse_sigmoid")
1562
+ class SparseSigmoid(Lambda):
1563
+ """Operation wrapping keras.ops.sparse_sigmoid."""
1564
+
1565
+ def __init__(self, **kwargs):
1566
+ try:
1567
+ super().__init__(func=keras.ops.sparse_sigmoid, **kwargs)
1568
+ except AttributeError as e:
1569
+ raise MissingKerasOps("SparseSigmoid", "keras.ops.sparse_sigmoid") from e
1570
+
1571
+ @ops_registry("keras.ops.sparsemax")
1572
+ class Sparsemax(Lambda):
1573
+ """Operation wrapping keras.ops.sparsemax."""
1574
+
1575
+ def __init__(self, **kwargs):
1576
+ try:
1577
+ super().__init__(func=keras.ops.sparsemax, **kwargs)
1578
+ except AttributeError as e:
1579
+ raise MissingKerasOps("Sparsemax", "keras.ops.sparsemax") from e
1580
+
1581
+ @ops_registry("keras.ops.split")
1582
+ class Split(Lambda):
1583
+ """Operation wrapping keras.ops.split."""
1584
+
1585
+ def __init__(self, **kwargs):
1586
+ try:
1587
+ super().__init__(func=keras.ops.split, **kwargs)
1588
+ except AttributeError as e:
1589
+ raise MissingKerasOps("Split", "keras.ops.split") from e
1590
+
1591
+ @ops_registry("keras.ops.sqrt")
1592
+ class Sqrt(Lambda):
1593
+ """Operation wrapping keras.ops.sqrt."""
1594
+
1595
+ def __init__(self, **kwargs):
1596
+ try:
1597
+ super().__init__(func=keras.ops.sqrt, **kwargs)
1598
+ except AttributeError as e:
1599
+ raise MissingKerasOps("Sqrt", "keras.ops.sqrt") from e
1600
+
1601
+ @ops_registry("keras.ops.square")
1602
+ class Square(Lambda):
1603
+ """Operation wrapping keras.ops.square."""
1604
+
1605
+ def __init__(self, **kwargs):
1606
+ try:
1607
+ super().__init__(func=keras.ops.square, **kwargs)
1608
+ except AttributeError as e:
1609
+ raise MissingKerasOps("Square", "keras.ops.square") from e
1610
+
1611
+ @ops_registry("keras.ops.squareplus")
1612
+ class Squareplus(Lambda):
1613
+ """Operation wrapping keras.ops.squareplus."""
1614
+
1615
+ def __init__(self, **kwargs):
1616
+ try:
1617
+ super().__init__(func=keras.ops.squareplus, **kwargs)
1618
+ except AttributeError as e:
1619
+ raise MissingKerasOps("Squareplus", "keras.ops.squareplus") from e
1620
+
1621
+ @ops_registry("keras.ops.squeeze")
1622
+ class Squeeze(Lambda):
1623
+ """Operation wrapping keras.ops.squeeze."""
1624
+
1625
+ def __init__(self, **kwargs):
1626
+ try:
1627
+ super().__init__(func=keras.ops.squeeze, **kwargs)
1628
+ except AttributeError as e:
1629
+ raise MissingKerasOps("Squeeze", "keras.ops.squeeze") from e
1630
+
1631
+ @ops_registry("keras.ops.stack")
1632
+ class Stack(Lambda):
1633
+ """Operation wrapping keras.ops.stack."""
1634
+
1635
+ def __init__(self, **kwargs):
1636
+ try:
1637
+ super().__init__(func=keras.ops.stack, **kwargs)
1638
+ except AttributeError as e:
1639
+ raise MissingKerasOps("Stack", "keras.ops.stack") from e
1640
+
1641
+ @ops_registry("keras.ops.std")
1642
+ class Std(Lambda):
1643
+ """Operation wrapping keras.ops.std."""
1644
+
1645
+ def __init__(self, **kwargs):
1646
+ try:
1647
+ super().__init__(func=keras.ops.std, **kwargs)
1648
+ except AttributeError as e:
1649
+ raise MissingKerasOps("Std", "keras.ops.std") from e
1650
+
1651
+ @ops_registry("keras.ops.stft")
1652
+ class Stft(Lambda):
1653
+ """Operation wrapping keras.ops.stft."""
1654
+
1655
+ def __init__(self, **kwargs):
1656
+ try:
1657
+ super().__init__(func=keras.ops.stft, **kwargs)
1658
+ except AttributeError as e:
1659
+ raise MissingKerasOps("Stft", "keras.ops.stft") from e
1660
+
1661
+ @ops_registry("keras.ops.sum")
1662
+ class Sum(Lambda):
1663
+ """Operation wrapping keras.ops.sum."""
1664
+
1665
+ def __init__(self, **kwargs):
1666
+ try:
1667
+ super().__init__(func=keras.ops.sum, **kwargs)
1668
+ except AttributeError as e:
1669
+ raise MissingKerasOps("Sum", "keras.ops.sum") from e
1670
+
1671
+ @ops_registry("keras.ops.svd")
1672
+ class Svd(Lambda):
1673
+ """Operation wrapping keras.ops.svd."""
1674
+
1675
+ def __init__(self, **kwargs):
1676
+ try:
1677
+ super().__init__(func=keras.ops.svd, **kwargs)
1678
+ except AttributeError as e:
1679
+ raise MissingKerasOps("Svd", "keras.ops.svd") from e
1680
+
1681
+ @ops_registry("keras.ops.swapaxes")
1682
+ class Swapaxes(Lambda):
1683
+ """Operation wrapping keras.ops.swapaxes."""
1684
+
1685
+ def __init__(self, **kwargs):
1686
+ try:
1687
+ super().__init__(func=keras.ops.swapaxes, **kwargs)
1688
+ except AttributeError as e:
1689
+ raise MissingKerasOps("Swapaxes", "keras.ops.swapaxes") from e
1690
+
1691
+ @ops_registry("keras.ops.swish")
1692
+ class Swish(Lambda):
1693
+ """Operation wrapping keras.ops.swish."""
1694
+
1695
+ def __init__(self, **kwargs):
1696
+ try:
1697
+ super().__init__(func=keras.ops.swish, **kwargs)
1698
+ except AttributeError as e:
1699
+ raise MissingKerasOps("Swish", "keras.ops.swish") from e
1700
+
1701
+ @ops_registry("keras.ops.take")
1702
+ class Take(Lambda):
1703
+ """Operation wrapping keras.ops.take."""
1704
+
1705
+ def __init__(self, **kwargs):
1706
+ try:
1707
+ super().__init__(func=keras.ops.take, **kwargs)
1708
+ except AttributeError as e:
1709
+ raise MissingKerasOps("Take", "keras.ops.take") from e
1710
+
1711
+ @ops_registry("keras.ops.take_along_axis")
1712
+ class TakeAlongAxis(Lambda):
1713
+ """Operation wrapping keras.ops.take_along_axis."""
1714
+
1715
+ def __init__(self, **kwargs):
1716
+ try:
1717
+ super().__init__(func=keras.ops.take_along_axis, **kwargs)
1718
+ except AttributeError as e:
1719
+ raise MissingKerasOps("TakeAlongAxis", "keras.ops.take_along_axis") from e
1720
+
1721
+ @ops_registry("keras.ops.tan")
1722
+ class Tan(Lambda):
1723
+ """Operation wrapping keras.ops.tan."""
1724
+
1725
+ def __init__(self, **kwargs):
1726
+ try:
1727
+ super().__init__(func=keras.ops.tan, **kwargs)
1728
+ except AttributeError as e:
1729
+ raise MissingKerasOps("Tan", "keras.ops.tan") from e
1730
+
1731
+ @ops_registry("keras.ops.tanh")
1732
+ class Tanh(Lambda):
1733
+ """Operation wrapping keras.ops.tanh."""
1734
+
1735
+ def __init__(self, **kwargs):
1736
+ try:
1737
+ super().__init__(func=keras.ops.tanh, **kwargs)
1738
+ except AttributeError as e:
1739
+ raise MissingKerasOps("Tanh", "keras.ops.tanh") from e
1740
+
1741
+ @ops_registry("keras.ops.tanh_shrink")
1742
+ class TanhShrink(Lambda):
1743
+ """Operation wrapping keras.ops.tanh_shrink."""
1744
+
1745
+ def __init__(self, **kwargs):
1746
+ try:
1747
+ super().__init__(func=keras.ops.tanh_shrink, **kwargs)
1748
+ except AttributeError as e:
1749
+ raise MissingKerasOps("TanhShrink", "keras.ops.tanh_shrink") from e
1750
+
1751
+ @ops_registry("keras.ops.threshold")
1752
+ class Threshold(Lambda):
1753
+ """Operation wrapping keras.ops.threshold."""
1754
+
1755
+ def __init__(self, **kwargs):
1756
+ try:
1757
+ super().__init__(func=keras.ops.threshold, **kwargs)
1758
+ except AttributeError as e:
1759
+ raise MissingKerasOps("Threshold", "keras.ops.threshold") from e
1760
+
1761
+ @ops_registry("keras.ops.tile")
1762
+ class Tile(Lambda):
1763
+ """Operation wrapping keras.ops.tile."""
1764
+
1765
+ def __init__(self, **kwargs):
1766
+ try:
1767
+ super().__init__(func=keras.ops.tile, **kwargs)
1768
+ except AttributeError as e:
1769
+ raise MissingKerasOps("Tile", "keras.ops.tile") from e
1770
+
1771
+ @ops_registry("keras.ops.top_k")
1772
+ class TopK(Lambda):
1773
+ """Operation wrapping keras.ops.top_k."""
1774
+
1775
+ def __init__(self, **kwargs):
1776
+ try:
1777
+ super().__init__(func=keras.ops.top_k, **kwargs)
1778
+ except AttributeError as e:
1779
+ raise MissingKerasOps("TopK", "keras.ops.top_k") from e
1780
+
1781
+ @ops_registry("keras.ops.trace")
1782
+ class Trace(Lambda):
1783
+ """Operation wrapping keras.ops.trace."""
1784
+
1785
+ def __init__(self, **kwargs):
1786
+ try:
1787
+ super().__init__(func=keras.ops.trace, **kwargs)
1788
+ except AttributeError as e:
1789
+ raise MissingKerasOps("Trace", "keras.ops.trace") from e
1790
+
1791
+ @ops_registry("keras.ops.transpose")
1792
+ class Transpose(Lambda):
1793
+ """Operation wrapping keras.ops.transpose."""
1794
+
1795
+ def __init__(self, **kwargs):
1796
+ try:
1797
+ super().__init__(func=keras.ops.transpose, **kwargs)
1798
+ except AttributeError as e:
1799
+ raise MissingKerasOps("Transpose", "keras.ops.transpose") from e
1800
+
1801
+ @ops_registry("keras.ops.tril")
1802
+ class Tril(Lambda):
1803
+ """Operation wrapping keras.ops.tril."""
1804
+
1805
+ def __init__(self, **kwargs):
1806
+ try:
1807
+ super().__init__(func=keras.ops.tril, **kwargs)
1808
+ except AttributeError as e:
1809
+ raise MissingKerasOps("Tril", "keras.ops.tril") from e
1810
+
1811
+ @ops_registry("keras.ops.triu")
1812
+ class Triu(Lambda):
1813
+ """Operation wrapping keras.ops.triu."""
1814
+
1815
+ def __init__(self, **kwargs):
1816
+ try:
1817
+ super().__init__(func=keras.ops.triu, **kwargs)
1818
+ except AttributeError as e:
1819
+ raise MissingKerasOps("Triu", "keras.ops.triu") from e
1820
+
1821
+ @ops_registry("keras.ops.trunc")
1822
+ class Trunc(Lambda):
1823
+ """Operation wrapping keras.ops.trunc."""
1824
+
1825
+ def __init__(self, **kwargs):
1826
+ try:
1827
+ super().__init__(func=keras.ops.trunc, **kwargs)
1828
+ except AttributeError as e:
1829
+ raise MissingKerasOps("Trunc", "keras.ops.trunc") from e
1830
+
1831
+ @ops_registry("keras.ops.unstack")
1832
+ class Unstack(Lambda):
1833
+ """Operation wrapping keras.ops.unstack."""
1834
+
1835
+ def __init__(self, **kwargs):
1836
+ try:
1837
+ super().__init__(func=keras.ops.unstack, **kwargs)
1838
+ except AttributeError as e:
1839
+ raise MissingKerasOps("Unstack", "keras.ops.unstack") from e
1840
+
1841
+ @ops_registry("keras.ops.var")
1842
+ class Var(Lambda):
1843
+ """Operation wrapping keras.ops.var."""
1844
+
1845
+ def __init__(self, **kwargs):
1846
+ try:
1847
+ super().__init__(func=keras.ops.var, **kwargs)
1848
+ except AttributeError as e:
1849
+ raise MissingKerasOps("Var", "keras.ops.var") from e
1850
+
1851
+ @ops_registry("keras.ops.view_as_complex")
1852
+ class ViewAsComplex(Lambda):
1853
+ """Operation wrapping keras.ops.view_as_complex."""
1854
+
1855
+ def __init__(self, **kwargs):
1856
+ try:
1857
+ super().__init__(func=keras.ops.view_as_complex, **kwargs)
1858
+ except AttributeError as e:
1859
+ raise MissingKerasOps("ViewAsComplex", "keras.ops.view_as_complex") from e
1860
+
1861
+ @ops_registry("keras.ops.view_as_real")
1862
+ class ViewAsReal(Lambda):
1863
+ """Operation wrapping keras.ops.view_as_real."""
1864
+
1865
+ def __init__(self, **kwargs):
1866
+ try:
1867
+ super().__init__(func=keras.ops.view_as_real, **kwargs)
1868
+ except AttributeError as e:
1869
+ raise MissingKerasOps("ViewAsReal", "keras.ops.view_as_real") from e
1870
+
1871
+ @ops_registry("keras.ops.zeros_like")
1872
+ class ZerosLike(Lambda):
1873
+ """Operation wrapping keras.ops.zeros_like."""
1874
+
1875
+ def __init__(self, **kwargs):
1876
+ try:
1877
+ super().__init__(func=keras.ops.zeros_like, **kwargs)
1878
+ except AttributeError as e:
1879
+ raise MissingKerasOps("ZerosLike", "keras.ops.zeros_like") from e
1880
+
1881
+ @ops_registry("keras.ops.image.affine_transform")
1882
+ class AffineTransform(Lambda):
1883
+ """Operation wrapping keras.ops.image.affine_transform."""
1884
+
1885
+ def __init__(self, **kwargs):
1886
+ try:
1887
+ super().__init__(func=keras.ops.image.affine_transform, **kwargs)
1888
+ except AttributeError as e:
1889
+ raise MissingKerasOps("AffineTransform", "keras.ops.image.affine_transform") from e
1890
+
1891
+ @ops_registry("keras.ops.image.crop_images")
1892
+ class CropImages(Lambda):
1893
+ """Operation wrapping keras.ops.image.crop_images."""
1894
+
1895
+ def __init__(self, **kwargs):
1896
+ try:
1897
+ super().__init__(func=keras.ops.image.crop_images, **kwargs)
1898
+ except AttributeError as e:
1899
+ raise MissingKerasOps("CropImages", "keras.ops.image.crop_images") from e
1900
+
1901
+ @ops_registry("keras.ops.image.elastic_transform")
1902
+ class ElasticTransform(Lambda):
1903
+ """Operation wrapping keras.ops.image.elastic_transform."""
1904
+
1905
+ def __init__(self, **kwargs):
1906
+ try:
1907
+ super().__init__(func=keras.ops.image.elastic_transform, **kwargs)
1908
+ except AttributeError as e:
1909
+ raise MissingKerasOps("ElasticTransform", "keras.ops.image.elastic_transform") from e
1910
+
1911
+ @ops_registry("keras.ops.image.extract_patches")
1912
+ class ExtractPatches(Lambda):
1913
+ """Operation wrapping keras.ops.image.extract_patches."""
1914
+
1915
+ def __init__(self, **kwargs):
1916
+ try:
1917
+ super().__init__(func=keras.ops.image.extract_patches, **kwargs)
1918
+ except AttributeError as e:
1919
+ raise MissingKerasOps("ExtractPatches", "keras.ops.image.extract_patches") from e
1920
+
1921
+ @ops_registry("keras.ops.image.gaussian_blur")
1922
+ class GaussianBlur(Lambda):
1923
+ """Operation wrapping keras.ops.image.gaussian_blur."""
1924
+
1925
+ def __init__(self, **kwargs):
1926
+ try:
1927
+ super().__init__(func=keras.ops.image.gaussian_blur, **kwargs)
1928
+ except AttributeError as e:
1929
+ raise MissingKerasOps("GaussianBlur", "keras.ops.image.gaussian_blur") from e
1930
+
1931
+ @ops_registry("keras.ops.image.hsv_to_rgb")
1932
+ class HsvToRgb(Lambda):
1933
+ """Operation wrapping keras.ops.image.hsv_to_rgb."""
1934
+
1935
+ def __init__(self, **kwargs):
1936
+ try:
1937
+ super().__init__(func=keras.ops.image.hsv_to_rgb, **kwargs)
1938
+ except AttributeError as e:
1939
+ raise MissingKerasOps("HsvToRgb", "keras.ops.image.hsv_to_rgb") from e
1940
+
1941
+ @ops_registry("keras.ops.image.pad_images")
1942
+ class PadImages(Lambda):
1943
+ """Operation wrapping keras.ops.image.pad_images."""
1944
+
1945
+ def __init__(self, **kwargs):
1946
+ try:
1947
+ super().__init__(func=keras.ops.image.pad_images, **kwargs)
1948
+ except AttributeError as e:
1949
+ raise MissingKerasOps("PadImages", "keras.ops.image.pad_images") from e
1950
+
1951
+ @ops_registry("keras.ops.image.perspective_transform")
1952
+ class PerspectiveTransform(Lambda):
1953
+ """Operation wrapping keras.ops.image.perspective_transform."""
1954
+
1955
+ def __init__(self, **kwargs):
1956
+ try:
1957
+ super().__init__(func=keras.ops.image.perspective_transform, **kwargs)
1958
+ except AttributeError as e:
1959
+ raise MissingKerasOps("PerspectiveTransform", "keras.ops.image.perspective_transform") from e
1960
+
1961
+ @ops_registry("keras.ops.image.resize")
1962
+ class Resize(Lambda):
1963
+ """Operation wrapping keras.ops.image.resize."""
1964
+
1965
+ def __init__(self, **kwargs):
1966
+ try:
1967
+ super().__init__(func=keras.ops.image.resize, **kwargs)
1968
+ except AttributeError as e:
1969
+ raise MissingKerasOps("Resize", "keras.ops.image.resize") from e
1970
+
1971
+ @ops_registry("keras.ops.image.rgb_to_grayscale")
1972
+ class RgbToGrayscale(Lambda):
1973
+ """Operation wrapping keras.ops.image.rgb_to_grayscale."""
1974
+
1975
+ def __init__(self, **kwargs):
1976
+ try:
1977
+ super().__init__(func=keras.ops.image.rgb_to_grayscale, **kwargs)
1978
+ except AttributeError as e:
1979
+ raise MissingKerasOps("RgbToGrayscale", "keras.ops.image.rgb_to_grayscale") from e
1980
+
1981
+ @ops_registry("keras.ops.image.rgb_to_hsv")
1982
+ class RgbToHsv(Lambda):
1983
+ """Operation wrapping keras.ops.image.rgb_to_hsv."""
1984
+
1985
+ def __init__(self, **kwargs):
1986
+ try:
1987
+ super().__init__(func=keras.ops.image.rgb_to_hsv, **kwargs)
1988
+ except AttributeError as e:
1989
+ raise MissingKerasOps("RgbToHsv", "keras.ops.image.rgb_to_hsv") from e