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
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 ortensor.item<T>()in C++ to convert a 0-dim tensor to a number