All the codes are giving errors
siddagra opened this issue · comments
None of the codes working, all give errors:
RuntimeError: output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
IndexError: invalid index of a 0-dim tensor. Use tensor.item()
in Python or tensor.item<T>()
in C++ to convert a 0-dim tensor to a number