Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 41 of 41 for new_axis_mask (0.14 sec)

  1. tensorflow/cc/gradients/array_grad_test.cc

      // y.shape = [2, 3, 3];
      RunTest(x, x_shape, y, {2, 3, 3});
    
      // y = [tf.newaxis, 2:6:2, 1:3, 1:3]
      y = StridedSlice(scope_, x, {0, 2, 1, 1}, {0, 6, 3, 3}, {1, 2, 1, 1},
                       StridedSlice::NewAxisMask(1 << 0));
      // y.shape = [1, 2, 2, 2];
      RunTest(x, x_shape, y, {1, 2, 2, 2});
    }
    
    TEST_F(ArrayGradTest, SliceGrad) {
      TensorShape x_shape({3, 5, 3});
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 19.3K bytes
    - Viewed (0)
Back to top