张量的访问元素

import numpy as np

# 创建一个二维张量
tensor = np.array([[1, 2, 3], [4, 5, 6]])

# 访问第一个元素（行号为0，列号为0）
element = tensor[0, 0]
print(element)  # 输出：1

import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;

// 创建一个二维张量
INDArray tensor = Nd4j.create(new double[][]{{1, 2, 3}, {4, 5, 6}});

// 访问第一个元素（行号为0，列号为0）
double element = tensor.getDouble(0, 0);
System.out.println(element);  // 输出：1.0

#include <iostream>
#include <tensorflow/core/framework/tensor.h>

using namespace tensorflow;

int main() {
  // 创建一个二维张量
  Tensor tensor(DT_FLOAT, TensorShape({2, 3}));
  auto tensor_map = tensor.tensor<float, 2>();

  // 设置张量元素的值
  tensor_map(0, 0) = 1.0;
  tensor_map(0, 1) = 2.0;
  tensor_map(0, 2) = 3.0;
  tensor_map(1, 0) = 4.0;
  tensor_map(1, 1) = 5.0;
  tensor_map(1, 2) = 6.0;

  // 访问第一个元素（行号为0，列号为0）
  float element = tensor_map(0, 0);
  std::cout << element << std::endl;  // 输出：1.0

  return 0;
}