When you need to create a new tensor, use type_as . This will make your code scale to any arbitrary number of GPUs or TPUs with Lightning. # before lightning ...
確定! 回上一頁