Assign different gradient weights

[Cris-chen]

How to write the code to assign different gradient weights to different classes of SparseCategoricalCrossentropy

[calmiLovesAI]

The sample_weight param may work. https://tensorflow.google.cn/api_docs/python/tf/keras/losses/SparseCategoricalCrossentropy#__call__