我开始使用Deep Java Library及其底层的数组操作包:ndarray。
问题非常简单。我想修改现有NDArray的第i
个元素,但无法做到这一点。我如何将第i
个元素设置为特定值?
文档中提到了许多set
方法。
以下是我尝试过的最小可复现示例:
var manager = NDManager.newBaseManager();var y = manager.create(new float[] {1, 2, 3, 4, 5});System.out.println("y before modification: " + y);y.set(new float[] {1, 100, 3, 4, 5});System.out.println("y after setting the entire array: " + y);// the following throws: "java.lang.UnsupportedOperationException: Tensor cannot be modified after creation"y.set(new NDIndex("1"), 1000f);System.out.println("y after setting the 1st element to 1000: " + y);
抛出的错误是:
java.lang.UnsupportedOperationException: Tensor cannot be modified after creation
回答:
您遇到的异常是因为您使用的是TensorFlow引擎。TensorFlow引擎不支持在创建后修改NDArray。
您可以选择PyTorch引擎。您的代码应该可以正常工作。您只需要在项目中将tensorflow依赖替换为pytorch。如果您的类路径中有多个引擎,您可以明确选择PyTorch:
var manager = NDManager.newBaseManager(Device.cpu(), "PyTorch");